From 46e15cd7a9b74ec716ea19a82acb8c79b403f75b Mon Sep 17 00:00:00 2001
From: "gregory.ashton" <gregory.ashton@ligo.org>
Date: Tue, 11 Apr 2017 11:02:50 +0200
Subject: [PATCH] Fixes corner_plot prior for rescaled params
---
pyfstat/mcmc_based_searches.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/pyfstat/mcmc_based_searches.py b/pyfstat/mcmc_based_searches.py
index d5c9bc7..263422d 100644
--- a/pyfstat/mcmc_based_searches.py
+++ b/pyfstat/mcmc_based_searches.py
@@ -571,7 +571,7 @@ class MCMCSearch(core.BaseSearchClass):
fig.subplots_adjust(hspace=0.05, wspace=0.05)
if add_prior:
- self.add_prior_to_corner(axes, samples_plt)
+ self.add_prior_to_corner(axes, self.samples)
fig_triangle.savefig('{}/{}_corner.png'.format(
self.outdir, self.label), dpi=dpi)
@@ -587,8 +587,8 @@ class MCMCSearch(core.BaseSearchClass):
subtractor = self.get_rescale_subtractor_for_key(key)
ax2 = ax.twinx()
ax2.get_yaxis().set_visible(False)
- ax2.plot(x, [(prior(xi)-subtractor)*multiplier for xi in x], '-r')
- ax.set_xlim(xlim)
+ ax2.plot((x-subtractor)*multiplier, [prior(xi) for xi in x], '-r')
+ ax2.set_xlim(xlim)
def plot_prior_posterior(self, normal_stds=2):
""" Plot the posterior in the context of the prior """
--
GitLab