Type is_nonzero, is_zero, wrap_in_cse

inducer · Oct 28, 2024 · 802b5aa · 802b5aa
1 parent 0e790f1
commit 802b5aa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pymbolic/primitives.py b/pymbolic/primitives.py
@@ -1890,7 +1890,7 @@ def unregister_constant_class(class_):
     VALID_CONSTANT_CLASSES = tuple(tmp)
 
 
-def is_nonzero(value):
+def is_nonzero(value: object) -> bool:
     if value is None:
         raise ValueError("is_nonzero is undefined for None")
 
@@ -1900,11 +1900,11 @@ def is_nonzero(value):
         return True
 
 
-def is_zero(value):
+def is_zero(value: object) -> bool:
     return not is_nonzero(value)
 
 
-def wrap_in_cse(expr, prefix=None):
+def wrap_in_cse(expr: ExpressionT, prefix=None) -> ExpressionT:
     if isinstance(expr, (Variable, Subscript)):
         return expr