From 7eff1a0c556be35cf0dae84d68ee644da16a9e75 Mon Sep 17 00:00:00 2001
From: "gregory.ashton" <gregory.ashton@ligo.org>
Date: Mon, 28 May 2018 11:33:19 +0000
Subject: [PATCH] Minor bug fixes

---
 pyfstat/mcmc_based_searches.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pyfstat/mcmc_based_searches.py b/pyfstat/mcmc_based_searches.py
index 789da34..c039605 100644
--- a/pyfstat/mcmc_based_searches.py
+++ b/pyfstat/mcmc_based_searches.py
@@ -2157,6 +2157,7 @@ class MCMCFollowUpSearch(MCMCSemiCoherentSearch):
         """
 
         self.nsegs = 1
+        self._set_likelihoodcoef()
         self._initiate_search_object()
         run_setup = self.init_run_setup(
             run_setup, NstarMax=NstarMax, Nsegs0=Nsegs0, log_table=log_table,
@@ -2241,7 +2242,8 @@ class MCMCFollowUpSearch(MCMCSemiCoherentSearch):
         self.lnprobs = lnprobs
         self.lnlikes = lnlikes
         self.all_lnlikelihood = all_lnlikelihood
-        self._save_data(sampler, samples, lnprobs, lnlikes, all_lnlikelihood)
+        self._save_data(sampler, samples, lnprobs, lnlikes, all_lnlikelihood,
+                        sampler.chain)
 
         if create_plots:
             try:
-- 
GitLab