Skip to content
Snippets Groups Projects
Commit f9e92e6a authored by Gregory Ashton's avatar Gregory Ashton
Browse files

Merge branch 'gw_conversion_join_mp_pool' into 'master'

Add pool.join after pool.close in bilby.gw.conversion

See merge request !1084
parents e04d1190 60e8f77f
No related branches found
No related tags found
1 merge request!1084Add pool.join after pool.close in bilby.gw.conversion
Pipeline #362417 passed
......@@ -33,6 +33,7 @@ James A Clark
Jeremy G Baier
John Veitch
Joshua Brandt
Josh Willis
Katerina Chatziioannou
Kaylee de Soto
Khun Sang Phukon
......
......@@ -1167,6 +1167,7 @@ def compute_snrs(sample, likelihood, npool=1):
)
new_samples = pool.map(_compute_snrs, tqdm(fill_args, file=sys.stdout))
pool.close()
pool.join()
else:
new_samples = [_compute_snrs(xx) for xx in tqdm(fill_args, file=sys.stdout)]
......@@ -1307,6 +1308,7 @@ def generate_posterior_samples_from_marginalized_likelihood(
if pool is not None:
pool.close()
pool.join()
new_samples = np.concatenate(
[np.array(val) for key, val in cached_samples_dict.items() if key != "_samples"]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment