From 7cf9da7134d094b98dba09621ac1d0728c5793c8 Mon Sep 17 00:00:00 2001
From: Colm Talbot <talbotcolm@gmail.com>
Date: Tue, 28 Feb 2023 11:39:56 -0500
Subject: [PATCH] TEST: make sure run test returns valid values
---
bilby/core/sampler/dynesty.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/bilby/core/sampler/dynesty.py b/bilby/core/sampler/dynesty.py
index 8abba033d..252462296 100644
--- a/bilby/core/sampler/dynesty.py
+++ b/bilby/core/sampler/dynesty.py
@@ -902,6 +902,7 @@ class Dynesty(NestedSampler):
import pandas as pd
self._set_sampling_method()
+ self._setup_pool()
self.sampler = self.sampler_init(
loglikelihood=_log_likelihood_wrapper,
prior_transform=_prior_transform_wrapper,
@@ -916,6 +917,8 @@ class Dynesty(NestedSampler):
self.pbar = tqdm(file=sys.stdout, initial=self.sampler.it)
self.sampler.run_nested(**sampler_kwargs)
+ self._close_pool()
+
if self.pbar is not None:
self.pbar = self.pbar.close()
print("")
--
GitLab