Update constant_fold.py

Author: BobTheBuidler

mypyc/irbuild/constant_fold.py

@@ -73,16 +73,15 @@ def constant_fold_expr(builder: IRBuilder, expr: Expression) -> ConstantValue |
         value = constant_fold_expr(builder, expr.expr)
         if value is not None and not isinstance(value, bytes):
             return constant_fold_unary_op(expr.op, value)
-    elif isinstance(expr, CallExpr) and isinstance(callee := expr.callee, MemberExpr):
-        # --- str.format constant folding
-        if callee.name == "format":
-            folded_args: list[ConstantValue] = []
-            for arg in expr.args:
-                arg_val = constant_fold_expr(arg, cur_mod_id)
-                if arg_val is None:
-                    return None
-                folded_args.append(arg_val)
-            return folded_callee.format(*folded_args)
+    # --- str.format constant folding
+    elif isinstance(expr, CallExpr) and isinstance(callee := expr.callee, MemberExpr) and callee.name == "format" and (folded_callee := constant_fold_expr(callee)) is not None:
+        folded_args: list[ConstantValue] = []
+        for arg in expr.args:
+            arg_val = constant_fold_expr(builder, arg)
+            if arg_val is None:
+                return None
+            folded_args.append(arg_val)
+        return folded_callee.format(*folded_args)
     return None