Skip to content

Commit d8974e8

Browse files
committed
narrow on initial assignment
1 parent 9b54ab1 commit d8974e8

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

mypy/checker.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3006,7 +3006,9 @@ def visit_assignment_stmt(self, s: AssignmentStmt) -> None:
30063006
# as X | Y.
30073007
if not (s.is_alias_def and self.is_stub):
30083008
with self.enter_final_context(s.is_final_def):
3009-
self.check_assignment(s.lvalues[-1], s.rvalue, s.type is None, s.new_syntax)
3009+
self.check_assignment(
3010+
s.lvalues[-1], s.rvalue, infer_lvalue_type=True, new_syntax=s.new_syntax
3011+
)
30103012

30113013
if s.is_alias_def:
30123014
self.check_type_alias_rvalue(s)
@@ -3034,7 +3036,7 @@ def visit_assignment_stmt(self, s: AssignmentStmt) -> None:
30343036
rvalue = self.temp_node(self.lookup_type(s.rvalue), s)
30353037
for lv in s.lvalues[:-1]:
30363038
with self.enter_final_context(s.is_final_def):
3037-
self.check_assignment(lv, rvalue, s.type is None)
3039+
self.check_assignment(lv, rvalue, infer_lvalue_type=True)
30383040

30393041
self.check_final(s)
30403042
if (

0 commit comments

Comments
 (0)