fix graph/basic.py

pymc-devs · Aug 14, 2024 · d5efa11 · d5efa11
1 parent 962772f
commit d5efa11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytensor/graph/basic.py b/pytensor/graph/basic.py
@@ -710,7 +710,7 @@ def clone(self, **kwargs):
         return cp
 
 
-class NominalVariable(AtomicVariable[_TypeType]):
+class NominalVariable(Generic[_TypeType, _IdType], AtomicVariable[_TypeType]):
     """A variable that enables alpha-equivalent comparisons."""
 
     __instances__: dict[tuple["Type", Hashable], "NominalVariable"] = {}