minor

inducer · Nov 12, 2024 · ad08e3e · ad08e3e
1 parent dd419b0
commit ad08e3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/loopy/statistics.py b/loopy/statistics.py
@@ -729,7 +729,7 @@ def __post_init__(self):
             from loopy.types import to_loopy_type
             object.__setattr__(self, "dtype", to_loopy_type(self.dtype))
 
-        assert isinstance(self.op_type, OpType) or self.op_type is None, self.op_type
+        assert isinstance(self.op_type, (OpType, type(None)))
 
         if not isinstance(self.count_granularity, (CountGranularity, type(None))):
             raise ValueError(