From 9688ee7c5b8adda9afc5a7400042a1387eef2f0e Mon Sep 17 00:00:00 2001 From: Ricardo Date: Fri, 16 Jul 2021 12:24:49 +0200 Subject: [PATCH] Revert changes in TestSMC::test_sample due to old pickling issues --- pymc3/tests/test_smc.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/pymc3/tests/test_smc.py b/pymc3/tests/test_smc.py index 20ca838724d..78b0b1544d8 100644 --- 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)