diff --git a/bilby/core/sampler/dynesty.py b/bilby/core/sampler/dynesty.py index 2f4644b5183d568fb630b728f7d4403222e66c5a..688efc417721fa0d9b600b5e011ccf668fe2770b 100644 --- a/bilby/core/sampler/dynesty.py +++ b/bilby/core/sampler/dynesty.py @@ -288,7 +288,6 @@ class Dynesty(NestedSampler): self.use_ratio ) ) - self.kwargs["pool"] = self.pool else: _initialize_global_variables( likelihood=self.likelihood, @@ -297,6 +296,7 @@ class Dynesty(NestedSampler): use_ratio=self.use_ratio ) self.pool = None + self.kwargs["pool"] = self.pool def _close_pool(self): if getattr(self, "pool", None) is not None: @@ -304,6 +304,7 @@ class Dynesty(NestedSampler): self.pool.close() self.pool.join() self.pool = None + self.kwargs["pool"] = self.pool logger.info("Finished closing worker pool.") def run_sampler(self):