Set model seed correctly in pymc3.tests.test_ndarray_backend

pymc-devs · Mar 23, 2021 · 20ddb4c · 20ddb4c
1 parent 3e5d206
commit 20ddb4c
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/pymc3/tests/test_ndarray_backend.py b/pymc3/tests/test_ndarray_backend.py
@@ -267,14 +267,12 @@ def test_sample_posterior_predictive(self, tmpdir_factory):
 
         assert save_dir == directory
 
-        seed = 10
-        np.random.seed(seed)
-        with TestSaveLoad.model():
+        with TestSaveLoad.model() as model:
+            model.default_rng.get_value(borrow=True).seed(10)
             ppc = pm.sample_posterior_predictive(self.trace)
 
-        seed = 10
-        np.random.seed(seed)
-        with TestSaveLoad.model():
+        with TestSaveLoad.model() as model:
+            model.default_rng.get_value(borrow=True).seed(10)
             trace2 = pm.load_trace(directory)
             ppc2 = pm.sample_posterior_predictive(trace2)
             ppc2f = pm.sample_posterior_predictive(trace2)