diff options
| author | shivesh <s.p.mandalia@qmul.ac.uk> | 2018-11-26 21:46:18 -0600 |
|---|---|---|
| committer | shivesh <s.p.mandalia@qmul.ac.uk> | 2018-11-26 21:46:18 -0600 |
| commit | 2a19e060fc9e982d6d4249a0e9eba0e4d6ac860d (patch) | |
| tree | 148095cdfb4e9c71728e89a632c5da69b230fb78 /utils | |
| parent | cc69e1c33d69a39367630a613b523bc15b38583f (diff) | |
| download | GolemFlavor-2a19e060fc9e982d6d4249a0e9eba0e4d6ac860d.tar.gz GolemFlavor-2a19e060fc9e982d6d4249a0e9eba0e4d6ac860d.zip | |
Mon 26 Nov 21:46:18 CST 2018
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/enums.py | 4 | ||||
| -rw-r--r-- | utils/gf.py | 19 |
2 files changed, 18 insertions, 5 deletions
diff --git a/utils/enums.py b/utils/enums.py index 2450ff7..22f91b8 100644 --- a/utils/enums.py +++ b/utils/enums.py @@ -10,6 +10,10 @@ Define Enums for the BSM flavour ratio analysis from enum import Enum +def str_enum(x): + return '{0}'.format(str(x).split('.')[-1]) + + class DataType(Enum): REAL = 1 ASIMOV = 2 diff --git a/utils/gf.py b/utils/gf.py index dfab69a..1998484 100644 --- a/utils/gf.py +++ b/utils/gf.py @@ -20,7 +20,7 @@ except: pass from utils.enums import DataType, Likelihood, SteeringCateg -from utils.misc import enum_parse, thread_factors +from utils.misc import enum_parse, parse_bool, thread_factors from utils.param import ParamSet @@ -61,8 +61,10 @@ def steering_params(args): steering_categ = args.ast params = gf.SteeringParams(gf.sampleTag.MagicTau) params.quiet = False - params.fastmode = True - # params.fastmode = False + if args.debug: + params.fastmode = False + else: + params.fastmode = True params.simToLoad= steering_categ.name.lower() params.evalThreads = args.threads # params.evalThreads = thread_factors(args.threads)[1] @@ -75,8 +77,12 @@ def steering_params(args): # For Tianlu # params.years = [999] - params.minFitEnergy = args.binning[0] # GeV - params.maxFitEnergy = args.binning[-1] # GeV + if hasattr(args, 'binning'): + params.minFitEnergy = args.binning[0] # GeV + params.maxFitEnergy = args.binning[-1] # GeV + else: + params.minFitEnergy = 6E4 # GeV + params.maxFitEnergy = 1E7 # GeV params.load_data_from_text_file = False return params @@ -140,6 +146,9 @@ def data_distributions(fitter): def gf_argparse(parser): parser.add_argument( + '--debug', default='False', type=parse_bool, help='Run without fastmode' + ) + parser.add_argument( '--data', default='asimov', type=partial(enum_parse, c=DataType), choices=DataType, help='select datatype' ) |