From 5af6880df87493d0b90cd3dbc9b3a1afe6052140 Mon Sep 17 00:00:00 2001 From: Pauli Virtanen Date: Thu, 30 May 2019 16:29:22 +0300 Subject: BENCH: fix benchmark suite importability on Numpy<1.17 --- benchmarks/benchmarks/bench_random.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'benchmarks') 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', -- cgit v1.2.1