Use discriminated union for nodetype (#272)

cms-nanoAOD · Jan 15, 2025 · 78f94f6 · 78f94f6
1 parent b07c2b4
commit 78f94f6
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/src/correctionlib/schemav2.py b/src/correctionlib/schemav2.py
@@ -70,13 +70,18 @@ def __rich__(self) -> str:
 
 # py3.7+: ForwardRef can be used instead of strings
 Content = Union[
-    "Binning",
-    "MultiBinning",
-    "Category",
-    "Formula",
-    "FormulaRef",
-    "Transform",
-    "HashPRNG",
+    Annotated[
+        Union[
+            "Binning",
+            "MultiBinning",
+            "Category",
+            "Formula",
+            "FormulaRef",
+            "Transform",
+            "HashPRNG",
+        ],
+        Field(discriminator="nodetype"),
+    ],
     float,
 ]