fix: fix mypy errors and add mypy as pre-commit

Author: 34j

.gitignore

@@ -35,3 +35,5 @@ tmp/
 *.egg
 dist/
 .DS_STORE
+venv
+.venv

.pre-commit-config.yaml

@@ -48,3 +48,8 @@ repos:
     rev: 23.7.0
     hooks:
       - id: black
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.13.0
+    hooks:
+      - id: mypy
+        additional_dependencies: []

pyproject.toml

@@ -32,3 +32,17 @@ build-backend = "setuptools.build_meta"
 
 [tool.black]
 line-length = 88
+
+[tool.mypy]
+mypy_path = "src/array_api_stubs/_draft"
+exclude = [
+    "docs/",
+    "spec/",
+    "venv/",
+    ".venv/",
+    "src/array_api_stubs/_2021_12/",
+    "src/array_api_stubs/_2022_12/",
+    "src/array_api_stubs/_2023_12/",
+    "src/array_api_conf.py"
+]
+disable_error_code = "empty-body,type-var"

src/array_api_stubs/_draft/__init__.py

@@ -1,6 +1,5 @@
 """Function stubs and API documentation for the array API standard."""
 
-from .array_object import *
 from .constants import *
 from .creation_functions import *
 from .data_type_functions import *

src/array_api_stubs/_draft/_types.py

@@ -46,9 +46,11 @@
 )
 from enum import Enum
 
-array = TypeVar("array", bound="array_")
+array = TypeVar("array", bound="_array")
 device = TypeVar("device")
 dtype = TypeVar("dtype")
+Device = TypeVar("Device")
+Dtype = TypeVar("Dtype")
 SupportsDLPack = TypeVar("SupportsDLPack")
 SupportsBufferProtocol = TypeVar("SupportsBufferProtocol")
 PyCapsule = TypeVar("PyCapsule")
@@ -88,7 +90,7 @@ def __len__(self, /) -> int:
         ...
 
 
-class Info(Protocol):
+class Info(Protocol[device]):
     """Namespace returned by `__array_namespace_info__`."""
 
     def capabilities(self) -> Capabilities:
@@ -147,12 +149,12 @@ def dtypes(
 )
 
 
-class _array(Protocol[array, dtype, device]):
+class _array(Protocol[array, Dtype, Device, PyCapsule]):  # type: ignore
     def __init__(self: array) -> None:
         """Initialize the attributes for the array object class."""
 
     @property
-    def dtype(self: array) -> dtype:
+    def dtype(self: array) -> Dtype:
         """
         Data type of the array elements.
 
@@ -163,7 +165,7 @@ def dtype(self: array) -> dtype:
         """
 
     @property
-    def device(self: array) -> device:
+    def device(self: array) -> Device:
         """
         Hardware device the array data resides on.
 
@@ -625,7 +627,7 @@ def __dlpack_device__(self: array, /) -> Tuple[Enum, int]:
               ONE_API = 14
         """
 
-    def __eq__(self: array, other: Union[int, float, bool, array], /) -> array:
+    def __eq__(self: array, other: Union[int, float, bool, array], /) -> array:  # type: ignore
         r"""
         Computes the truth value of ``self_i == other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -1072,7 +1074,7 @@ def __mul__(self: array, other: Union[int, float, array], /) -> array:
             Added complex data type support.
         """
 
-    def __ne__(self: array, other: Union[int, float, bool, array], /) -> array:
+    def __ne__(self: array, other: Union[int, float, bool, array], /) -> array:  # type: ignore
         """
         Computes the truth value of ``self_i != other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -1342,7 +1344,7 @@ def __xor__(self: array, other: Union[int, bool, array], /) -> array:
         """
 
     def to_device(
-        self: array, device: device, /, *, stream: Optional[Union[int, Any]] = None
+        self: array, device: Device, /, *, stream: Optional[Union[int, Any]] = None
     ) -> array:
         """
         Copy the array from the device on which it currently resides to the specified ``device``.

src/array_api_stubs/_draft/linalg.py

@@ -301,7 +301,7 @@ def matrix_norm(
     /,
     *,
     keepdims: bool = False,
-    ord: Optional[Union[int, float, Literal[inf, -inf, "fro", "nuc"]]] = "fro",
+    ord: Optional[Union[int, float, Literal[inf, -inf, "fro", "nuc"]]] = "fro",  # type: ignore
 ) -> array:
     """
     Computes the matrix norm of a matrix (or a stack of matrices) ``x``.
@@ -781,7 +781,7 @@ def trace(x: array, /, *, offset: int = 0, dtype: Optional[dtype] = None) -> arr
     """
 
 
-def vecdot(x1: array, x2: array, /, *, axis: int = None) -> array:
+def vecdot(x1: array, x2: array, /, *, axis: int | None = None) -> array:
     """Alias for :func:`~array_api.vecdot`."""
 
 
@@ -791,7 +791,7 @@ def vector_norm(
     *,
     axis: Optional[Union[int, Tuple[int, ...]]] = None,
     keepdims: bool = False,
-    ord: Union[int, float, Literal[inf, -inf]] = 2,
+    ord: Union[int, float, Literal[inf, -inf]] = 2,  # type: ignore
 ) -> array:
     r"""
     Computes the vector norm of a vector (or batch of vectors) ``x``.