Skip to content
Snippets Groups Projects
Commit a7741796 authored by Colm Talbot's avatar Colm Talbot
Browse files

Hide initialise function

parent c4db9452
No related branches found
No related tags found
1 merge request!754Allow dynesty to run with multiprocessing
Pipeline #115536 failed
......@@ -24,7 +24,7 @@ _priors = None
_search_parameter_keys = None
def initialize_likelihood_and_prior(likelihood, priors, search_parameter_keys):
def _initialize_global_variables(likelihood, priors, search_parameter_keys):
"""
Store a global copy of the likelihood, priors, and search keys for
multiprocessing.
......@@ -294,12 +294,12 @@ class Dynesty(NestedSampler):
import multiprocessing
self.pool = multiprocessing.Pool(
processes=self.kwargs["queue_size"],
initializer=initialize_likelihood_and_prior,
initializer=_initialize_global_variables,
initargs=(self.likelihood, self.priors, self._search_parameter_keys)
)
self.kwargs["pool"] = self.pool
else:
initialize_likelihood_and_prior(
_initialize_global_variables(
likelihood=self.likelihood,
priors=self.priors,
search_parameter_keys=self._search_parameter_keys
......
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