Skip to content

Commit d0a6e7c

Browse files
Update for_helpers.py
1 parent 293f001 commit d0a6e7c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mypyc/irbuild/for_helpers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
from __future__ import annotations
99

10-
from typing import Callable, ClassVar
10+
from typing import Callable, ClassVar, cast
1111

1212
from mypy.nodes import (
1313
ARG_POS,
@@ -1249,10 +1249,10 @@ def get_expr_length(builder: IRBuilder, expr: Expression) -> int | None:
12491249
if all(arg is not None for arg in arg_lengths):
12501250
return min(arg_lengths) # type: ignore [type-var]
12511251
elif fullname == "builtins.range" and len(expr.args) <= 3:
1252-
folded_args = [constant_fold_expr(builder, arg) for arg in args]
1252+
folded_args = [constant_fold_expr(builder, arg) for arg in expr.args]
12531253
if all(isinstance(arg, int) for arg in folded_args):
12541254
try:
1255-
return len(range(*folded_args))
1255+
return len(range(*cast(list[int], folded_args)))
12561256
except ValueError: # prevent crash if invalid args
12571257
pass
12581258

0 commit comments

Comments
 (0)