fixups

mitmproxy · Aug 27, 2024 · be98a83 · be98a83
1 parent 720521d
commit be98a83
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/pdoc/doc_pyi.py b/pdoc/doc_pyi.py
@@ -58,6 +58,7 @@ def _import_stub_file(module_name: str, stub_file: Path) -> types.ModuleType:
     try:
         loader = importlib.machinery.SourceFileLoader(module_name, str(stub_file))
         spec = importlib.util.spec_from_file_location(module_name, stub_file, loader=loader)
+        assert spec is not None
         m = importlib.util.module_from_spec(spec)
         loader.exec_module(m)
         return m

diff --git a/test/testdata/type_stubs/__init__.pyi b/test/testdata/type_stubs/__init__.pyi
@@ -1,6 +1,7 @@
 from typing import Any
 from typing import Iterable
 from typing import overload
+
 from ._utils import ImportedClass
 
 def func(x: str, y: Any, z: "Iterable[str]") -> int: ...
@@ -22,3 +23,10 @@ class Class:
     def overloaded(self, x: int) -> int: ...
     @overload
     def overloaded(self, x: str) -> str: ...
+
+__all__ = [
+    "func",
+    "var",
+    "Class",
+    "ImportedClass",
+]