Wed May 23 16:23:12 CDT 2018

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