Update expression.py

Author: BobTheBuidler

mypyc/irbuild/expression.py

@@ -120,7 +120,8 @@
 from mypyc.primitives.str_ops import str_slice_op
 from mypyc.primitives.tuple_ops import list_tuple_op, tuple_slice_op
 
-ConstantValueTuple = Union[ConstantValue, tuple["ConstantValueTuple", ...]]
+ConstantValueOrTuple = Union[ConstantValue, "ConstantValueTuple"]
+ConstantValueTuple = tuple[ConstantValueOrTuple, ...]
 
 # Name and attribute references
 
@@ -1014,7 +1015,7 @@ def _visit_tuple_display(builder: IRBuilder, expr: TupleExpr) -> Value:
 
 def dict_literal_values(
     builder: IRBuilder, items: Sequence[tuple[Expression | None, Expression]], line: int
-) -> dict[ConstantValueTuple, ConstantValueTuple] | None:
+) -> dict[ConstantValueOrTuple, ConstantValueOrTuple] | None:
     """Try to extract a constant dict from a dict literal, recursively staticizing nested dicts.
 
     If all keys and values are deeply immutable and constant (including nested dicts as values),
@@ -1023,15 +1024,13 @@ def dict_literal_values(
 
     def constant_fold_expr_or_tuple(
         builder: IRBuilder, expr: Expression
-    ) -> ConstantValueTuple | None:
+    ) -> ConstantValueOrTuple | None:
         value = constant_fold_expr(builder, expr)
         if value is not None:
             return value
         if not isinstance(expr, TupleExpr):
             return None
-        folded: ConstantValueTuple = tuple(
-            map(partial(constant_fold_expr_or_tuple, builder), expr.items)
-        )
+        folded = tuple(map(partial(constant_fold_expr_or_tuple, builder), expr.items))
         return folded if None not in folded else None
 
     result = {}