From 845c55c269a59620bbf8a6c0d8adab575e1185dc Mon Sep 17 00:00:00 2001 From: shivesh Date: Thu, 8 Nov 2018 15:02:27 -0600 Subject: Thu 8 Nov 15:02:27 CST 2018 --- fr.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'fr.py') diff --git a/fr.py b/fr.py index b550688..d960ab9 100755 --- a/fr.py +++ b/fr.py @@ -10,6 +10,7 @@ HESE BSM flavour ratio MCMC analysis script from __future__ import absolute_import, division +import os import argparse from functools import partial @@ -81,8 +82,8 @@ def nuisance_argparse(parser): def process_args(args): """Process the input args.""" - if args.fix_mixing is not MixingScenario.NONE and args.fix_scale: - raise NotImplementedError('Fixed mixing and scale not implemented') + # if args.fix_mixing is not MixingScenario.NONE and args.fix_scale: + # raise NotImplementedError('Fixed mixing and scale not implemented') if args.fix_mixing is not MixingScenario.NONE and args.fix_mixing_almost: raise NotImplementedError( '--fix-mixing and --fix-mixing-almost cannot be used together' @@ -165,6 +166,12 @@ def main(): llh_paramset, nwalkers=args.nwalkers ) + if args.save_measured_fr: + n = misc_utils.gen_identifier(args) + '.txt' + f = args.output_measured_fr + n + if os.path.isfile(f): + open(f, 'w').close() + samples = mcmc_utils.mcmc( p0 = p0, ln_prob = ln_prob, @@ -172,6 +179,7 @@ def main(): nwalkers = args.nwalkers, burnin = args.burnin, nsteps = args.nsteps, + args = args, threads = 1 # TODO(shivesh): broken because you cannot pickle a GolemFitPy object # threads = misc_utils.thread_factors(args.threads)[0] -- cgit v1.2.3