clean up prefold

tserg · Nov 14, 2023 · 78bad6c · 78bad6c
1 parent 0bfaf2f
commit 78bad6c
Showing 1 changed file with 1 addition and 16 deletions.
diff --git a/vyper/semantics/analysis/pre_typecheck.py b/vyper/semantics/analysis/pre_typecheck.py
@@ -46,22 +46,7 @@ def pre_typecheck(node: vy_ast.Module):
 
 
 def prefold(node: vy_ast.VyperNode, constants: dict) -> None:
-    if isinstance(node, vy_ast.BinOp):
-        node._metadata["folded_value"] = node.prefold()
-
-    if isinstance(node, vy_ast.UnaryOp):
-        node._metadata["folded_value"] = node.prefold()
-
-    if isinstance(node, vy_ast.Compare):
-        node._metadata["folded_value"] = node.prefold()
-
-    if isinstance(node, vy_ast.BoolOp):
-        node._metadata["folded_value"] = node.prefold()
-
-    if isinstance(node, vy_ast.Subscript):
-        node._metadata["folded_value"] = node.prefold()
-
-    if isinstance(node, vy_ast.List):
+    if isinstance(node, (vy_ast.BinOp, vy_ast.BoolOp, vy_ast.Compare, vy_ast.List, vy_ast.Subscript, vy_ast.UnaryOp)):
         node._metadata["folded_value"] = node.prefold()
 
     if isinstance(node, vy_ast.Name):