Skip to content

Commit 2c389b5

Browse files
committed
fix: Register top-module in collection earlier
This is needed when we inspect compiled submodules on the spot (non-discoverable on disk) *and* they are merged with found stubs.
1 parent 2f05967 commit 2c389b5

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/_griffe/loader.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -506,7 +506,6 @@ def stats(self) -> Stats:
506506

507507
def _load_package(self, package: Package | NamespacePackage, *, submodules: bool = True) -> Module:
508508
top_module = self._load_module(package.name, package.path, submodules=submodules)
509-
self.modules_collection.set_member(top_module.path, top_module)
510509
if isinstance(package, NamespacePackage):
511510
return top_module
512511
if package.stubs:
@@ -558,6 +557,8 @@ def _load_module_path(
558557
module = self._inspect_module(module_name, module_path, parent)
559558
else:
560559
raise LoadingError("Cannot load compiled module without inspection")
560+
if parent is None:
561+
self.modules_collection.set_member(module.path, module)
561562
if submodules:
562563
self._load_submodules(module)
563564
return module

0 commit comments

Comments
 (0)