Skip to content

Commit 8d23615

Browse files
authored
Merge pull request numpy#26638 from ngoldbaum/reproducible-benchmarks
MNT: use reproducible RNG sequences in benchmarks
2 parents 24e6811 + 72026ec commit 8d23615

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

benchmarks/benchmarks/bench_clip.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@ class ClipFloat(Benchmark):
1111
]
1212

1313
def setup(self, dtype, size):
14-
rng = np.random.default_rng()
15-
self.array = rng.random(size=size).astype(dtype)
14+
rnd = np.random.RandomState(994584855)
15+
self.array = rnd.random(size=size).astype(dtype)
1616
self.dataout = np.full_like(self.array, 0.5)
1717

1818
def time_clip(self, dtype, size):
@@ -27,8 +27,8 @@ class ClipInteger(Benchmark):
2727
]
2828

2929
def setup(self, dtype, size):
30-
rng = np.random.default_rng()
31-
self.array = rng.integers(256, size=size, dtype=dtype)
30+
rnd = np.random.RandomState(1301109903)
31+
self.array = rnd.randint(256, size=size, dtype=dtype)
3232
self.dataout = np.full_like(self.array, 128)
3333

3434
def time_clip(self, dtype, size):

benchmarks/benchmarks/bench_function_base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,8 @@ def random(size, dtype):
159159
Returns a randomly-shuffled array.
160160
"""
161161
arr = np.arange(size, dtype=dtype)
162-
np.random.shuffle(arr)
162+
rnd = np.random.RandomState(1792364059)
163+
rnd.shuffle(arr)
163164
return arr
164165

165166
@staticmethod

0 commit comments

Comments
 (0)