From 8b6aaa4f84acd67c9d66374ade5110e4fc28080b Mon Sep 17 00:00:00 2001
From: BobTheBuidler <70677534+BobTheBuidler@users.noreply.github.com>
Date: Thu, 2 Oct 2025 17:04:07 -0400
Subject: [PATCH] [mypyc] feat: support constant folding in
 `transform_assignment_expr`

We can't yet fold the expression itself, but we can fold its value expr

This PR implements folding for the value expr
---
 mypyc/irbuild/expression.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mypyc/irbuild/expression.py b/mypyc/irbuild/expression.py
index 59ecc4ac2c5c..a84c786bbd8e 100644
--- a/mypyc/irbuild/expression.py
+++ b/mypyc/irbuild/expression.py
@@ -1097,7 +1097,7 @@ def transform_generator_expr(builder: IRBuilder, o: GeneratorExpr) -> Value:
 
 
 def transform_assignment_expr(builder: IRBuilder, o: AssignmentExpr) -> Value:
-    value = builder.accept(o.value)
+    value = try_constant_fold(builder, o.value) or builder.accept(o.value)
     target = builder.get_assignment_target(o.target)
     builder.assign(target, value, o.line)
     return value