@@ -92,7 +92,7 @@ def constant_fold_expr(builder: IRBuilder, expr: Expression) -> ConstantValue |
92
92
if folded_items is not None and all (
93
93
isinstance (item , str ) for item in folded_items
94
94
):
95
- return folded_callee .join (folded_items )
95
+ return folded_callee .join (folded_items ) # type: ignore [arg-type]
96
96
97
97
# builtins.bytes methods
98
98
elif isinstance (folded_callee , bytes ):
@@ -107,7 +107,7 @@ def constant_fold_expr(builder: IRBuilder, expr: Expression) -> ConstantValue |
107
107
if folded_items is not None and all (
108
108
isinstance (item , bytes ) for item in folded_items
109
109
):
110
- return folded_callee .join (folded_items )
110
+ return folded_callee .join (folded_items ) # type: ignore [arg-type]
111
111
return None
112
112
113
113
@@ -143,11 +143,11 @@ def constant_fold_container_expr(
143
143
builder : IRBuilder , expr : ListExpr | TupleExpr
144
144
) -> list [ConstantValue ] | tuple [ConstantValue , ...] | None :
145
145
folded_items = [constant_fold_expr (builder , item_expr ) for item_expr in expr .items ]
146
- if all ( isinstance ( item , CONST_TYPES ) for item in folded_items ) :
147
- if isinstance ( expr , ListExpr ):
148
- return folded_items
149
- elif isinstance ( expr , TupleExpr ):
150
- return tuple ( folded_items )
151
- else :
152
- raise NotImplementedError ( type ( expr ), expr )
153
- return None
146
+ if None in folded_items :
147
+ return None
148
+ if isinstance ( expr , ListExpr ):
149
+ return folded_items # type: ignore [return-value]
150
+ elif isinstance ( expr , TupleExpr ):
151
+ return tuple ( folded_items ) # type: ignore [arg-type]
152
+ else :
153
+ raise NotImplementedError ( type ( expr ), expr )
0 commit comments