add a constant folding step after ast rewriting

dakk · Jul 8, 2024 · af1d013 · af1d013
1 parent a44e60e
commit af1d013
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/qlasskit/ast2ast/ast2ast.py b/qlasskit/ast2ast/ast2ast.py
@@ -49,5 +49,8 @@ def ast2ast(a_tree):
     # Rewrite the ast
     a_tree = ASTRewriter().visit(a_tree)
 
+    # Fold constants again
+    a_tree = ConstantFolder().visit(a_tree)
+
     # print(ast.dump(a_tree))
     return a_tree