diff --git a/pyfstat/mcmc_based_searches.py b/pyfstat/mcmc_based_searches.py index d5c9bc708439a11579832704f7ddb734d15d3bca..263422d3593ee0f94e2cfcdb2c6f2f6b83371bfc 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 """