Sometimes coefficient has a different name

cms-analysis · Nov 2, 2023 · 6ca7196 · 6ca7196
1 parent 484a45e
commit 6ca7196
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/InterferenceModels.py b/python/InterferenceModels.py
@@ -172,11 +172,11 @@ def done(self):
             if isinstance(histfunc, ROOT.CMSHistFunc):
                 histfunc.injectExternalMorph(func)
             elif isinstance(histfunc, ROOT.CMSHistSum):
-                coefname = "n_exp_final_bin{channel}_proc_{process}".format(channel=channel, process=process)
+                postfix = "bin{channel}_proc_{process}".format(channel=channel, process=process)
                 for idx, coef in enumerate(histfunc.coefList()):
-                    if coef.GetName() == coefname:
+                    if coef.GetName() in ("n_exp_" + postfix, "n_exp_final_" + postfix):
                         if self.verbose:
-                            print("Injecting external morph for " + coefname)
+                            print("Injecting external morph for " + funcname)
                         histfunc.injectExternalMorph(idx, func)
                         break