Fix some test failures

GridTools · Feb 11, 2025 · ed0248e · ed0248e
1 parent 101a25b
commit ed0248e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/gt4py/next/iterator/transforms/constant_folding.py b/src/gt4py/next/iterator/transforms/constant_folding.py
@@ -25,6 +25,11 @@ def value_from_literal(literal: ir.Literal):
 
 
 class UndoCanonicalizeMinus(eve.NodeTranslator):
+    PRESERVED_ANNEX_ATTRS = (
+        "type",
+        "domain",
+    )
+
     def visit_FunCall(self, node: ir.FunCall, **kwargs) -> ir.Node:
         node = super().generic_visit(node, **kwargs)
         # `a + (-b)` -> `a - b` , `-a + b` -> `b - a`, `-a + (-b)` -> `-a - b`