BENCH: fix benchmark suite importability on Numpy<1.17

jazcap53 · May 30, 2019 · 5af6880 · 5af6880
1 parent ec3cb9d
commit 5af6880
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/benchmarks/benchmarks/bench_random.py b/benchmarks/benchmarks/bench_random.py
@@ -4,7 +4,13 @@
 
 import numpy as np
 
-from numpy.random import RandomState, Generator
+from numpy.random import RandomState
+
+try:
+    from numpy.random import Generator
+except ImportError:
+    pass
+
 
 class Random(Benchmark):
     params = ['normal', 'uniform', 'weibull 1', 'binomial 10 0.5',