Do not include seeded_test fixture in exported BaseTestDistributionRa…

…ndom
pymc-devs · Aug 8, 2023 · d59a960 · d59a960
1 parent ccad4c8
commit d59a960
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pymc/testing.py b/pymc/testing.py
@@ -848,7 +848,7 @@ class BaseTestDistributionRandom:
     repeated_params_shape = 5
     random_state = None
 
-    def test_distribution(self, seeded_test):
+    def test_distribution(self):
         self.validate_tests_list()
         if self.pymc_dist == pm.Wishart:
             with pytest.warns(UserWarning, match="can currently not be used for MCMC sampling"):

diff --git a/tests/distributions/test_continuous.py b/tests/distributions/test_continuous.py
@@ -1825,7 +1825,7 @@ class TestStudentT(BaseTestDistributionRandom):
 
 class TestHalfStudentT(BaseTestDistributionRandom):
     def halfstudentt_rng_fn(self, df, loc, scale, size, rng):
-        return np.abs(st.t.rvs(df=df, loc=loc, scale=scale, size=size))
+        return np.abs(st.t.rvs(df=df, loc=loc, scale=scale, size=size, random_state=rng))
 
     pymc_dist = pm.HalfStudentT
     pymc_dist_params = {"nu": 5.0, "sigma": 2.0}

diff --git a/tests/distributions/test_multivariate.py b/tests/distributions/test_multivariate.py
@@ -1972,7 +1972,7 @@ class TestKroneckerNormal(BaseTestDistributionRandom):
     def kronecker_rng_fn(self, size, mu, covs=None, sigma=None, rng=None):
         cov = pm.math.kronecker(covs[0], covs[1]).eval()
         cov += sigma**2 * np.identity(cov.shape[0])
-        return st.multivariate_normal.rvs(mean=mu, cov=cov, size=size)
+        return st.multivariate_normal.rvs(mean=mu, cov=cov, size=size, random_state=rng)
 
     pymc_dist = pm.KroneckerNormal