diff options
Diffstat (limited to 'utils/gf.py')
| -rw-r--r-- | utils/gf.py | 14 |
1 files changed, 12 insertions, 2 deletions
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 |