From 77fc28427d96808e74d028697ae8122c900c9ee5 Mon Sep 17 00:00:00 2001 From: Rong Ou Date: Thu, 20 Jun 2019 16:51:48 -0700 Subject: [PATCH] fix benchmark_tree.py (#4593) --- tests/benchmark/benchmark_tree.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/benchmark/benchmark_tree.py b/tests/benchmark/benchmark_tree.py index 446deb714..380e03463 100644 --- a/tests/benchmark/benchmark_tree.py +++ b/tests/benchmark/benchmark_tree.py @@ -22,7 +22,7 @@ def run_benchmark(args): if not (dtest.num_row() == args.rows * args.test_size and dtrain.num_row() == args.rows * (1 - args.test_size)): raise ValueError("Wrong rows") - except ValueError: + except: print("Generating dataset: {} rows * {} columns".format(args.rows, args.columns)) print("{}/{} test/train split".format(args.test_size, 1.0 - args.test_size)) tmp = time.time()