check histpdf size in ShapeTools.py

cms-analysis · Jun 23, 2024 · fcef577 · fcef577
1 parent 6a0cbf3
commit fcef577
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/ShapeTools.py b/python/ShapeTools.py
@@ -1005,11 +1005,11 @@ def getPdf(self, channel, process, _cache={}):
                     pdfs.Add(self.shape2Pdf(shapeUp, channel, process))
                     pdfs.Add(self.shape2Pdf(shapeDown, channel, process))
                 histpdf = nominalPdf if nominalPdf.InheritsFrom("RooDataHist") else nominalPdf.dataHist()
-                xvar = histpdf.get().first()
-                yvar = histpdf.get().second()
                 if histpdf.get().getSize() > 2:
                     raise ValueError("No support for 3+ dimensional histpdfs")
-                elif yvar:
+                elif histpdf.get().getSize() > 1:
+                    xvar = histpdf.get().first()
+                    yvar = histpdf.get().second()
                     rhp = ROOT.FastVerticalInterpHistPdf2D2(
                         "shape%s_%s_%s_morph" % (postFix, channel, process),
                         "",
@@ -1022,6 +1022,7 @@ def getPdf(self, channel, process, _cache={}):
                         qalgo,
                     )
                 else:
+                    xvar = histpdf.get().first()
                     rhp = ROOT.FastVerticalInterpHistPdf2(
                         "shape%s_%s_%s_morph" % (postFix, channel, process),
                         "",