Set a seed for a flaky pymc3.tests.test_sampling test

pymc-devs · Mar 28, 2021 · a3164e7 · a3164e7
1 parent c744775
commit a3164e7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pymc3/tests/test_sampling.py b/pymc3/tests/test_sampling.py
@@ -967,6 +967,8 @@ def test_layers(self):
             a = pm.Uniform("a", lower=0, upper=1, size=10)
             b = pm.Binomial("b", n=1, p=a, size=10)
 
+        model.default_rng.get_value(borrow=True).seed(232093)
+
         b_sampler = aesara.function([], b)
         avg = np.stack([b_sampler() for i in range(10000)]).mean(0)
         npt.assert_array_almost_equal(avg, 0.5 * np.ones((10,)), decimal=2)