fix benchmark_tree.py (#4593)

2019-06-20 16:51:48 -07:00 · 2019-06-20 16:51:48 -07:00 · 77fc28427d
commit 77fc28427d
parent 9494950ee7
1 changed files with 1 additions and 1 deletions
--- 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()