Avoid unnecessary imports to numpy.typing

Author: sarlinpe

pybind11_stubgen/parser/mixins/fix.py

@@ -168,6 +168,11 @@ def _add_import(self, name: QualifiedName) -> None:
         if module_name is None:
             self.report_error(NameResolutionError(name))
             return
+        # Corner case: we know that annotations in numpy.typing.* are accessible from
+        # numpy and we want to avoid unnecessary imports once these annotations are
+        # replaced by subsequent numpy fixers.
+        if module_name == ("numpy", "typing"):
+            module_name = module_name.parent
         self.__extra_imports.add(Import(name=None, origin=module_name))
 
     def _get_parent_module(self, name: QualifiedName) -> QualifiedName | None:

tests/stubs/python-3.12/pybind11-v3.0.0/numpy-array-use-type-var/demo/_bindings/eigen.pyi

@@ -3,7 +3,6 @@ from __future__ import annotations
 import typing
 
 import numpy
-import numpy.typing
 import scipy.sparse
 
 __all__: list[str] = [

tests/stubs/python-3.12/pybind11-v3.0.0/numpy-array-use-type-var/demo/_bindings/numpy.pyi

@@ -3,7 +3,6 @@ from __future__ import annotations
 import typing
 
 import numpy
-import numpy.typing
 
 __all__: list[str] = [
     "accept_ndarray_float64",