summaryrefslogtreecommitdiff
path: root/benchmarks
diff options
context:
space:
mode:
Diffstat (limited to 'benchmarks')
-rw-r--r--benchmarks/benchmarks/bench_random.py8
1 files changed, 7 insertions, 1 deletions
diff --git a/benchmarks/benchmarks/bench_random.py b/benchmarks/benchmarks/bench_random.py
index c9c7baf7c..621b72d3a 100644
--- a/benchmarks/benchmarks/bench_random.py
+++ b/benchmarks/benchmarks/bench_random.py
@@ -4,7 +4,13 @@ from .common import Benchmark
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',