diff --git a/tupak/core/result.py b/tupak/core/result.py index a5440ef102de95df6985299feef642ca1a12421f..161d0fecb1aedb3b39352de73b7a2e825ad4d9df 100644 --- a/tupak/core/result.py +++ b/tupak/core/result.py @@ -102,7 +102,7 @@ class Result(dict): except Exception as e: logging.error( "\n\n Saving the data has failed with the following message:\n {} \n\n" - .format(e)) + .format(e)) def save_posterior_samples(self): filename = '{}/{}_posterior_samples.txt'.format(self.outdir, self.label) @@ -166,6 +166,10 @@ class Result(dict): if parameters is None: parameters = self.search_parameter_keys + if 'lionize' in kwargs and kwargs['lionize'] is True: + defaults_kwargs['truth_color'] = 'tab:blue' + defaults_kwargs['color'] = '#FF8C00' + xs = self.posterior[parameters].values kwargs['labels'] = kwargs.get( 'labels', self.get_latex_labels_from_parameter_keys( @@ -229,7 +233,7 @@ class Result(dict): self.posterior['chi_eff'] = (self.posterior.a_1 * np.cos(self.posterior.tilt_1) + self.posterior.q * self.posterior.a_2 * np.cos(self.posterior.tilt_2)) / ( - 1 + self.posterior.q) + 1 + self.posterior.q) self.posterior['chi_p'] = max(self.posterior.a_1 * np.sin(self.posterior.tilt_1), (4 * self.posterior.q + 3) / (3 * self.posterior.q + 4) * self.posterior.q * self.posterior.a_2 * np.sin(self.posterior.tilt_2)) @@ -248,4 +252,3 @@ class Result(dict): elif typeA in [np.ndarray]: return np.all(A == B) return False -