Seed benchmark tests

aesara-devs · Jan 7, 2023 · 0f8d849 · 0f8d849 · github-actions · Jan 7, 2023
1 parent 445b489
commit 0f8d849
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/link/jax/test_elemwise.py b/tests/link/jax/test_elemwise.py
@@ -111,7 +111,8 @@ def test_logsumexp_benchmark(size, axis, benchmark):
  X_max = at.switch(at.isinf(X_max), 0, X_max)
  X_lse = at.log(at.sum(at.exp(X - X_max), axis=axis, keepdims=True)) + X_max
 
- X_val = np.random.normal(size=size)
+ rng = np.random.default_rng(23920)
+ X_val = rng.normal(size=size)
 
  X_lse_fn = aesara.function([X], X_lse, mode="JAX")
 

diff --git a/tests/link/numba/test_elemwise.py b/tests/link/numba/test_elemwise.py
@@ -529,7 +529,8 @@ def test_logsumexp_benchmark(size, axis, benchmark):
  X_max = at.switch(at.isinf(X_max), 0, X_max)
  X_lse = at.log(at.sum(at.exp(X - X_max), axis=axis, keepdims=True)) + X_max
 
- X_val = np.random.normal(size=size)
+ rng = np.random.default_rng(23920)
+ X_val = rng.normal(size=size)
 
  X_lse_fn = aesara.function([X], X_lse, mode="NUMBA")