Fix dangling step in test_population

pymc-devs · Oct 7, 2024 · b810389 · b810389
1 parent cdcdb58
commit b810389
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/sampling/test_population.py b/tests/sampling/test_population.py
@@ -65,7 +65,7 @@ def test_nonparallelized_chains_are_random(self):
                     cores=1,
                     draws=20,
                     tune=0,
-                    step=DEMetropolis(),
+                    step=step,
                     compute_convergence_checks=False,
                 )
                 samples = idata.posterior["x"].values[:, 5]
@@ -82,7 +82,7 @@ def test_parallelized_chains_are_random(self):
                     cores=4,
                     draws=20,
                     tune=0,
-                    step=DEMetropolis(),
+                    step=step,
                     compute_convergence_checks=False,
                 )
                 samples = idata.posterior["x"].values[:, 5]