backend ereased before initializing new EMCEE object

sibirrer · Sep 3, 2024 · f482350 · f482350
1 parent 3a5a45b
commit f482350
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/hierarc/Sampling/mcmc_sampling.py b/hierarc/Sampling/mcmc_sampling.py
@@ -40,9 +40,6 @@ def get_emcee_sampler(
         """
 
         num_param = self.param.num_param
-        sampler = emcee.EnsembleSampler(
-            n_walkers, num_param, self.chain.likelihood, args=(), **kwargs_emcee
-        )
         mean_start = self.param.kwargs2args(**kwargs_mean_start)
         sigma_start = self.param.kwargs2args(**kwargs_sigma_start)
         p0 = sampling_util.sample_ball(mean_start, sigma_start, n_walkers)
@@ -52,6 +49,9 @@ def get_emcee_sampler(
                 p0 = None
             else:
                 backend.reset(n_walkers, num_param)
+        sampler = emcee.EnsembleSampler(
+            n_walkers, num_param, self.chain.likelihood, args=(), **kwargs_emcee
+        )
         sampler.run_mcmc(p0, n_burn + n_run, progress=True)
 
         return sampler