Revert changes in TestSMC::test_sample due to old pickling issues

pymc-devs · Jul 23, 2021 · 9688ee7 · 9688ee7
1 parent 4a30ad6
commit 9688ee7
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/pymc3/tests/test_smc.py b/pymc3/tests/test_smc.py
@@ -75,12 +75,7 @@ def two_gaussians(x):
 
     def test_sample(self):
         with self.SMC_test:
-
-            mtrace = pm.sample_smc(
-                draws=self.samples,
-                cores=1,  # Fails in parallel due to #4799
-                return_inferencedata=False,
-            )
+            mtrace = pm.sample_smc(draws=self.samples, return_inferencedata=False)
 
         x = mtrace["X"]
         mu1d = np.abs(x).mean(axis=0)