diff --git a/pymc_experimental/tests/test_marginal_model.py b/pymc_experimental/tests/test_marginal_model.py index 4e455354..cab0404c 100644 --- a/pymc_experimental/tests/test_marginal_model.py +++ b/pymc_experimental/tests/test_marginal_model.py @@ -296,6 +296,8 @@ def test_not_supported_marginalized(): @pytest.mark.filterwarnings("error") def test_marginalized_deterministic_and_potential(): + rng = np.random.default_rng(299) + with MarginalModel() as m: x = pm.Bernoulli("x", p=0.7) y = pm.Normal("y", x) @@ -306,7 +308,7 @@ def test_marginalized_deterministic_and_potential(): with pytest.warns(UserWarning, match="There are multiple dependent variables"): m.marginalize([x]) - y_draw, z_draw, det_draw, pot_draw = pm.draw([y, z, det, pot], draws=5) + y_draw, z_draw, det_draw, pot_draw = pm.draw([y, z, det, pot], draws=5, random_seed=rng) np.testing.assert_almost_equal(y_draw + z_draw, det_draw) np.testing.assert_almost_equal(det_draw, pot_draw - 1)