Update crates/red_knot_python_semantic/src/types/infer.rs

Co-authored-by: Carl Meyer <carl@oddbird.net>
astral-sh · Dec 20, 2024 · 3b1bca6 · 3b1bca6
1 parent 092ba09
commit 3b1bca6
Showing 1 changed file with 9 additions and 13 deletions.
diff --git a/crates/red_knot_python_semantic/src/types/infer.rs b/crates/red_knot_python_semantic/src/types/infer.rs
@@ -1927,21 +1927,17 @@ impl<'db> TypeInferenceBuilder<'db> {
 
         if let Some(value) = value.as_deref() {
             let value_ty = self.infer_expression(value);
-            if self.file().is_stub(self.db().upcast()) {
-                self.add_declaration_with_binding(
-                    assignment.into(),
-                    definition,
-                    annotation_ty,
-                    Type::Unknown,
-                );
+            let value_ty = if self.file().is_stub(self.db().upcast()) && value.is_ellipsis_literal_expr() {
+                annotation_ty
             } else {
-                self.add_declaration_with_binding(
-                    assignment.into(),
-                    definition,
-                    annotation_ty,
-                    value_ty,
-                );
+                value_ty
             }
+            self.add_declaration_with_binding(
+                assignment.into(),
+                definition,
+                annotation_ty,
+                value_ty,
+            );
         } else {
             self.add_declaration(assignment.into(), definition, annotation_ty);
         }