Seed sampling in TestNutsCheckTrace.test_linalg

pymc-devs · May 24, 2022 · c0e4f08 · c0e4f08
1 parent 2b987d8
commit c0e4f08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc/tests/test_step.py b/pymc/tests/test_step.py
@@ -523,7 +523,7 @@ def test_linalg(self, caplog):
             b = at.slinalg.solve(floatX(np.eye(2)), a, check_finite=False)
             Normal("c", mu=b, size=2, initval=floatX(np.r_[0.0, 0.0]))
             caplog.clear()
-            trace = sample(20, tune=5, chains=2, return_inferencedata=False)
+            trace = sample(20, tune=5, chains=2, return_inferencedata=False, random_seed=526)
             warns = [msg.msg for msg in caplog.records]
             assert np.any(trace["diverging"])
             assert (