diff --git a/bilby/core/sampler/dynesty.py b/bilby/core/sampler/dynesty.py index 3d8c6cd8568e6158e9314ef6973f91facb1c3c8f..c1604c6fd2578fe70c16a7918eff92104a506017 100644 --- a/bilby/core/sampler/dynesty.py +++ b/bilby/core/sampler/dynesty.py @@ -546,6 +546,7 @@ class Dynesty(NestedSampler): self.sampler.versions = dict( bilby=bilby_version, dynesty=dynesty_version ) + del self.sampler.pool, self.sampler.M if dill.pickles(self.sampler): safe_file_dump(self.sampler, self.resume_file, dill) logger.info("Written checkpoint file {}".format(self.resume_file)) @@ -554,6 +555,12 @@ class Dynesty(NestedSampler): "Cannot write pickle resume file! " "Job will not resume if interrupted." ) + self.sampler.pool = self.pool + if self.sampler.pool is not None: + self.sampler.M = self.sampler.pool.M + else: + self.sampler.M = map + def plot_current_state(self): if self.check_point_plot: