Skip to content

Commit b84bfbd

Browse files
fix mypy errs
1 parent 3a063f0 commit b84bfbd

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

mypyc/irbuild/constant_fold.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)