Skip to content

Commit e8f5c79

Browse files
committed
narrow on initial assignment
1 parent 2ebc690 commit e8f5c79

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
@@ -2998,7 +2998,9 @@ def visit_assignment_stmt(self, s: AssignmentStmt) -> None:
29982998
# as X | Y.
29992999
if not (s.is_alias_def and self.is_stub):
30003000
with self.enter_final_context(s.is_final_def):
3001-
self.check_assignment(s.lvalues[-1], s.rvalue, s.type is None, s.new_syntax)
3001+
self.check_assignment(
3002+
s.lvalues[-1], s.rvalue, infer_lvalue_type=True, new_syntax=s.new_syntax
3003+
)
30023004

30033005
if s.is_alias_def:
30043006
self.check_type_alias_rvalue(s)
@@ -3026,7 +3028,7 @@ def visit_assignment_stmt(self, s: AssignmentStmt) -> None:
30263028
rvalue = self.temp_node(self.lookup_type(s.rvalue), s)
30273029
for lv in s.lvalues[:-1]:
30283030
with self.enter_final_context(s.is_final_def):
3029-
self.check_assignment(lv, rvalue, s.type is None)
3031+
self.check_assignment(lv, rvalue, infer_lvalue_type=True)
30303032

30313033
self.check_final(s)
30323034
if (

0 commit comments

Comments
 (0)