Skip to content
Snippets Groups Projects
Commit bc030cfd authored by Gregory Ashton's avatar Gregory Ashton
Browse files

Make ignore keys settable

parent 16160d9c
No related branches found
No related tags found
1 merge request!750Improve ptemcee
......@@ -49,7 +49,7 @@ class Ptemcee(MCMCSampler):
autocorr_c=5, safety=1, frac_threshold=0.01,
autocorr_tol=50, autocorr_tau=5, min_tau=1, check_point_deltaT=600,
threads=1, exit_code=77, plot=False, store_walkers=False,
**kwargs):
ignore_keys_for_tau="recalib", **kwargs):
super(Ptemcee, self).__init__(
likelihood=likelihood, priors=priors, outdir=outdir,
label=label, use_ratio=use_ratio, plot=plot,
......@@ -73,6 +73,7 @@ class Ptemcee(MCMCSampler):
self.threads = threads
self.store_walkers = store_walkers
self.ignore_keys_for_tau = ignore_keys_for_tau
self.check_point_plot = check_point_plot
self.resume_file = "{}/{}_checkpoint_resume.pickle".format(self.outdir, self.label)
......@@ -172,7 +173,7 @@ class Ptemcee(MCMCSampler):
taus = []
for ii in range(sampler.nwalkers):
for jj, key in enumerate(self.search_parameter_keys):
if "recalib" in key:
if self.ignore_keys_for_tau and self.ignore_keys_for_tau in key:
continue
try:
taus.append(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment