Commit 6a8b89a9 authored by Colm Talbot's avatar Colm Talbot
Browse files

Fix checkpointing

parent 923d1563
Pipeline #115784 failed with stage
in 4 minutes and 51 seconds
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment