Use callable to test callable

microsoft · Dec 23, 2024 · 9a4a58e · 9a4a58e
1 parent 834815b
commit 9a4a58e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxscript/rewriter/_ir_utils.py b/onnxscript/rewriter/_ir_utils.py
@@ -89,7 +89,7 @@ def is_singleton_value(
     scalar = get_singleton_value(val)
     if scalar is None:
         return False
-    if isinstance(expected, Callable):
+    if callable(expected):
         return expected(scalar)
     if isinstance(expected, int):
         return expected == scalar