diff --git a/bilby/core/sampler/pymultinest.py b/bilby/core/sampler/pymultinest.py index 960f2f4a3a231fc43f6f51538c9381d133341f5b..11b5f250165343415bc7db8009f2470f9e314244 100644 --- a/bilby/core/sampler/pymultinest.py +++ b/bilby/core/sampler/pymultinest.py @@ -129,9 +129,9 @@ class Pymultinest(NestedSampler): def _apply_multinest_boundaries(self): if self.kwargs["wrapped_params"] is None: - self.kwargs["wrapped_params"] = [] - for param, value in self.priors.items(): - if value.boundary == "periodic": + self.kwargs["wrapped_params"] = list() + for param in self.search_parameter_keys: + if self.priors[param].boundary == "periodic": self.kwargs["wrapped_params"].append(1) else: self.kwargs["wrapped_params"].append(0)