bool

microsoft · Jan 3, 2025 · 75a355d · 75a355d
1 parent 04ca4b5
commit 75a355d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxscript/optimizer/_constant_folding.py b/onnxscript/optimizer/_constant_folding.py
@@ -257,7 +257,7 @@ def _get_bool_value(val: ir.Value | None) -> bool | None:
     value = _get_numpy_value(val)
     if value is None:
         return None
-    if value.size == 1 and value.dtype is bool:
+    if value.size == 1 and value.dtype == np.bool_:
         return value.item(0)
     return None