diff --git a/tupak/core/sampler/dynesty.py b/tupak/core/sampler/dynesty.py index 61fa1a54d1d90e964a85ff636668b5d720d94d2f..dd84e473b9de6d1614ebe5d58614df736c911bed 100644 --- a/tupak/core/sampler/dynesty.py +++ b/tupak/core/sampler/dynesty.py @@ -131,11 +131,11 @@ class Dynesty(Sampler): # self.result.sampler_output = out weights = np.exp(out['logwt'] - out['logz'][-1]) - self.result.samples = dynesty.utils.resample_equal( - out.samples, weights) + self.result.samples = dynesty.utils.resample_equal(out.samples, weights) self.result.nested_samples = DataFrame( out.samples, columns=self.search_parameter_keys) self.result.nested_samples['weights'] = weights + self.result.nested_samples['log_likelihood'] = out.logl idxs = [np.unique(np.where(self.result.samples[ii] == out.samples)[0]) for ii in range(len(out.logl))] self.result.log_likelihood_evaluations = out.logl[idxs]