Address review comments

GridTools · Mar 2, 2025 · dfe819c · dfe819c
1 parent 9653694
commit dfe819c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gt4py/eve/visitors.py b/src/gt4py/eve/visitors.py
@@ -173,7 +173,7 @@ def generic_visit(self, node: concepts.RootNode, **kwargs: Any) -> Any:
                     if (new_child := self.visit(child, **kwargs)) is not NOTHING
                 }
             )
-            if self.PRESERVED_ANNEX_ATTRS and (old_annex := getattr(node, "__node_annex__", None)):
+            if self.PRESERVED_ANNEX_ATTRS and getattr(node, "__node_annex__", None):
                 _preserve_annex(node, new_node, self.PRESERVED_ANNEX_ATTRS)
 
             return new_node
@@ -207,7 +207,7 @@ def visit(self, node: concepts.RootNode, **kwargs: Any) -> Any:
             isinstance(node, concepts.Node)
             and isinstance(new_node, concepts.Node)
             and self.PRESERVED_ANNEX_ATTRS
-            and (old_annex := getattr(node, "__node_annex__", None))
+            and getattr(node, "__node_annex__", None)
         ):
             _preserve_annex(node, new_node, self.PRESERVED_ANNEX_ATTRS)