From cc4e70ccd0d249fb5585c16d932b52467aaff969 Mon Sep 17 00:00:00 2001 From: shivesh Date: Wed, 23 May 2018 16:23:12 -0500 Subject: Wed May 23 16:23:12 CDT 2018 --- utils/gf.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) (limited to 'utils/gf.py') diff --git a/utils/gf.py b/utils/gf.py index 06a6125..0d098f1 100644 --- a/utils/gf.py +++ b/utils/gf.py @@ -71,7 +71,7 @@ def steering_params(args): return params -def set_up_as(fitter, params): +def setup_asimov(fitter, params): print 'Injecting the model', params asimov_params = gf.FitParameters(gf.sampleTag.MagicTau) for parm in params: @@ -79,13 +79,23 @@ def set_up_as(fitter, params): fitter.SetupAsimov(asimov_params) +def setup_realisation(fitter, params): + print 'Injecting the model', params + asimov_params = gf.FitParameters(gf.sampleTag.MagicTau) + for parm in params: + asimov_params.__setattr__(parm.name, float(parm.value)) + fitter.Swallow(fitter.SpitExpectation(asimov_params)) + + def setup_fitter(args, asimov_paramset): datapaths = gf.DataPaths() sparams = steering_params(args) npp = gf.NewPhysicsParams() fitter = gf.GolemFit(datapaths, sparams, npp) if args.data is DataType.ASIMOV: - set_up_as(fitter, asimov_paramset) + setup_asimov(fitter, asimov_paramset) + elif args.data is DataType.REALISATION: + setup_realisation(fitter, asimov_paramset) elif args.data is DataType.REAL: print 'Using MagicTau DATA' return fitter -- cgit v1.2.3