Better error for missing input (#243)

cms-nanoAOD · Jun 18, 2024 · dfff1d8 · dfff1d8
1 parent aa4e4e0
commit dfff1d8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/correctionlib/schemav2.py b/src/correctionlib/schemav2.py
@@ -357,6 +357,11 @@ def summarize(
         self, nodecount: Dict[str, int], inputstats: Dict[str, _SummaryInfo]
     ) -> None:
         nodecount["Category"] += 1
+        if self.input not in inputstats:
+            raise RuntimeError(
+                f"The input variable {self.input} of a Category node is not defined "
+                "in the inputs of the Correction object"
+            )
         inputstats[self.input].values |= {item.key for item in self.content}
         inputstats[self.input].default |= self.default is not None
         for item in self.content: