diff --git a/bilby/core/sampler/dynesty.py b/bilby/core/sampler/dynesty.py index c1604c6fd2578fe70c16a7918eff92104a506017..ca460ee702038787c3ba554037e171e805f87152 100644 --- a/bilby/core/sampler/dynesty.py +++ b/bilby/core/sampler/dynesty.py @@ -546,7 +546,8 @@ class Dynesty(NestedSampler): self.sampler.versions = dict( bilby=bilby_version, dynesty=dynesty_version ) - del self.sampler.pool, self.sampler.M + self.sampler.pool = None + self.sampler.M = map if dill.pickles(self.sampler): safe_file_dump(self.sampler, self.resume_file, dill) logger.info("Written checkpoint file {}".format(self.resume_file)) @@ -557,9 +558,7 @@ class Dynesty(NestedSampler): ) self.sampler.pool = self.pool if self.sampler.pool is not None: - self.sampler.M = self.sampler.pool.M - else: - self.sampler.M = map + self.sampler.M = self.sampler.pool.map def plot_current_state(self):