mypy

Author: znicholls

src/example_fgen_basic/error_v/creation.py

@@ -19,9 +19,7 @@
 except (ModuleNotFoundError, ImportError) as exc:  # pragma: no cover
     raise CompiledExtensionNotFoundError("example_fgen_basic._lib.m_error_v_w") from exc
 try:
-    from example_fgen_basic._lib import (  # type: ignore
-        m_error_v_creation_w,
-    )
+    from example_fgen_basic._lib import m_error_v_creation_w
 except (ModuleNotFoundError, ImportError) as exc:  # pragma: no cover
     raise CompiledExtensionNotFoundError(
         "example_fgen_basic._lib.m_error_v_creation_w"

src/example_fgen_basic/error_v/passing.py

@@ -21,9 +21,7 @@
 except (ModuleNotFoundError, ImportError) as exc:  # pragma: no cover
     raise CompiledExtensionNotFoundError("example_fgen_basic._lib.m_error_v_w") from exc
 try:
-    from example_fgen_basic._lib import (  # type: ignore
-        m_error_v_passing_w,
-    )
+    from example_fgen_basic._lib import m_error_v_passing_w
 except (ModuleNotFoundError, ImportError) as exc:  # pragma: no cover
     raise CompiledExtensionNotFoundError(
         "example_fgen_basic._lib.m_error_v_passing_w"
@@ -84,7 +82,9 @@ def pass_errors(invs: tuple[ErrorV, ...]) -> NP_ARRAY_OF_BOOL:
     )
 
     # Convert the result to boolean
-    res_raw = m_error_v_passing_w.pass_errors(instance_indexes, n=instance_indexes.size)
+    res_raw: NP_ARRAY_OF_INT = m_error_v_passing_w.pass_errors(
+        instance_indexes, n=instance_indexes.size
+    )
     res = res_raw.astype(bool)
 
     # Tell Fortran to finalise the objects on the Fortran side

src/example_fgen_basic/typing.py

@@ -7,22 +7,23 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from typing import Any, TypeAlias, Union
+    from typing import Union
 
     import numpy as np
     import numpy.typing as npt
+    from typing_extensions import Any, TypeAlias
 
-    NP_ARRAY_OF_INT: TypeAlias = npt.NDArray[np.integer]
+    NP_ARRAY_OF_INT: TypeAlias = npt.NDArray[np.integer[Any]]
     """
     Type alias for an array of numpy int
     """
 
-    NP_ARRAY_OF_FLOAT: TypeAlias = npt.NDArray[np.floating]
+    NP_ARRAY_OF_FLOAT: TypeAlias = npt.NDArray[np.floating[Any]]
     """
     Type alias for an array of numpy floats
     """
 
-    NP_FLOAT_OR_INT: TypeAlias = Union[np.floating, np.integer]
+    NP_FLOAT_OR_INT: TypeAlias = Union[np.floating[Any], np.integer[Any]]
     """
     Type alias for a numpy float or int (not complex)
     """