From 1ebf9534aad966f1076808eb95d3901a0e3881df Mon Sep 17 00:00:00 2001
From: Gregory Ashton <gregory.ashton@ligo.org>
Date: Tue, 28 Feb 2017 13:59:49 +0100
Subject: [PATCH] Adds check of number of segments to MCMCSC search

---
 pyfstat/mcmc_based_searches.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pyfstat/mcmc_based_searches.py b/pyfstat/mcmc_based_searches.py
index dacfbe0..b9bec08 100644
--- a/pyfstat/mcmc_based_searches.py
+++ b/pyfstat/mcmc_based_searches.py
@@ -1432,6 +1432,14 @@ class MCMCSemiCoherentSearch(MCMCSearch):
 
         self.log_input()
 
+    def get_save_data_dictionary(self):
+        d = dict(nsteps=self.nsteps, nwalkers=self.nwalkers,
+                 ntemps=self.ntemps, theta_keys=self.theta_keys,
+                 theta_prior=self.theta_prior, scatter_val=self.scatter_val,
+                 log10temperature_min=self.log10temperature_min,
+                 BSGL=self.BSGL, nsegs=self.nsegs)
+        return d
+
     def inititate_search_object(self):
         logging.info('Setting up search object')
         self.search = SemiCoherentSearch(
-- 
GitLab