feat: Support inspecting packages (__init__ modules)

mkdocstrings · May 12, 2024 · 3f74f67 · 3f74f67
1 parent 964e0d2
commit 3f74f67
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/griffe/agents/inspector.py b/src/griffe/agents/inspector.py
@@ -176,6 +176,9 @@ def get_module(self, import_paths: Sequence[str | Path] | None = None) -> Module
             parent_path = self.filepath.parent
             for _ in range(import_path.count(".")):
                 parent_path = parent_path.parent
+            # Climb up one more time for `__init__` modules.
+            if self.filepath.stem == "__init__":
+                parent_path = parent_path.parent
             if parent_path not in import_paths:
                 import_paths.insert(0, parent_path)