@@ -92,7 +92,7 @@ def constant_fold_expr(builder: IRBuilder, expr: Expression) -> ConstantValue |
9292 if folded_items is not None and all (
9393 isinstance (item , str ) for item in folded_items
9494 ):
95- return folded_callee .join (folded_items )
95+ return folded_callee .join (folded_items ) # type: ignore [arg-type]
9696
9797 # builtins.bytes methods
9898 elif isinstance (folded_callee , bytes ):
@@ -107,7 +107,7 @@ def constant_fold_expr(builder: IRBuilder, expr: Expression) -> ConstantValue |
107107 if folded_items is not None and all (
108108 isinstance (item , bytes ) for item in folded_items
109109 ):
110- return folded_callee .join (folded_items )
110+ return folded_callee .join (folded_items ) # type: ignore [arg-type]
111111 return None
112112
113113
@@ -143,11 +143,11 @@ def constant_fold_container_expr(
143143 builder : IRBuilder , expr : ListExpr | TupleExpr
144144) -> list [ConstantValue ] | tuple [ConstantValue , ...] | None :
145145 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