diff --git a/pyfstat.py b/pyfstat.py index bb5fb3c1c7b1bafe83c78e48ce8c9d335730f184..7d7781cd692359cc8b9e58d83161815ea8a3ef7c 100755 --- a/pyfstat.py +++ b/pyfstat.py @@ -662,7 +662,7 @@ class MCMCSearch(BaseSearchClass): return p0 - def run(self): + def run(self, proposal_scale_factor=None): if self.old_data_is_okay_to_use is True: logging.warning('Using saved data from {}'.format( @@ -679,7 +679,7 @@ class MCMCSearch(BaseSearchClass): sampler = emcee.PTSampler( self.ntemps, self.nwalkers, self.ndim, self.logl, self.logp, logpargs=(self.theta_prior, self.theta_keys, self.search), - loglargs=(self.search,), betas=self.betas) + loglargs=(self.search,), betas=self.betas, a=proposal_scale_factor) p0 = self.generate_initial_p0() p0 = self.apply_corrections_to_p0(p0)