Fix infer domain ops

GridTools · Feb 20, 2025 · 2c14648 · 2c14648
1 parent bd8dbaa
commit 2c14648
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/gt4py/next/iterator/transforms/infer_domain_ops.py b/src/gt4py/next/iterator/transforms/infer_domain_ops.py
@@ -93,7 +93,7 @@ def visit_FunCall(self, node: ir.FunCall, **kwargs) -> ir.Node:
                 raise ValueError(f"{fun} is not a valid comparison operator.")
 
         if cpm.is_call_to(node, builtins.BINARY_LOGICAL_BUILTINS) and all(
-            isinstance(arg, (ir.Literal, ir.FunCall)) for arg in node.args
+            isinstance(arg.type, ts.DomainType) for arg in node.args
         ):
             if cpm.is_call_to(node, "and_"):
                 # TODO: domain promotion
@@ -102,8 +102,7 @@ def visit_FunCall(self, node: ir.FunCall, **kwargs) -> ir.Node:
                         *[domain_utils.SymbolicDomain.from_expr(arg) for arg in node.args]
                     ).as_expr()
                 )
-
             else:
-                raise NotImplementedError
+                raise NotImplementedError()
 
         return node