.

Author: AlexWaygood

crates/ty_python_semantic/src/module_name.rs

@@ -318,7 +318,7 @@ impl ModuleName {
         db: &dyn Db,
         importing_file: File,
     ) -> Result<Self, ModuleNameResolutionError> {
-        relative_module_name(db, importing_file, None, NonZeroU32::new(1).unwrap())
+        Self::from_identifier_parts(db, importing_file, None, 1)
     }
 }
 

crates/ty_python_semantic/src/semantic_index/builder.rs

@@ -1551,11 +1551,11 @@ impl<'ast> Visitor<'ast> for SemanticIndexBuilder<'_, 'ast> {
                                 );
                             }
                         } else {
-                            self.imported_modules.extend(
-                                module_name
-                                    .ancestors()
-                                    .zip(std::iter::repeat(ImportKind::ImportFrom)),
-                            );
+                            for name in module_name.ancestors() {
+                                self.imported_modules
+                                    .entry(name)
+                                    .or_insert(ImportKind::ImportFrom);
+                            }
                         }
                     }
                 }

crates/ty_python_semantic/src/types.rs

@@ -11724,13 +11724,9 @@ impl<'db> ModuleLiteralType<'db> {
             .into_iter()
             .flat_map(|file| imported_modules(db, file))
             .filter_map(|(submodule_name, kind)| {
-                Some((submodule_name.relative_to(self.module(db).name(db))?, kind))
-            })
-            .filter_map(|(relative_submodule, kind)| {
-                relative_submodule
-                    .components()
-                    .next()
-                    .map(|module| (Name::from(module), *kind))
+                let relative_name = submodule_name.relative_to(self.module(db).name(db))?;
+                let available_attribute = relative_name.components().next()?;
+                Some((Name::from(available_attribute), *kind))
             })
     }