Add device protocol to fix type ignore

Signed-off-by: nstarman <[email protected]>

Author: nstarman

src/array_api_stubs/_draft/_types.py

@@ -52,8 +52,16 @@
     from .array_object import Array
     from .data_types import DType
 
+
+class Device(Protocol):
+    """Protocol for device objects."""
+
+    def __eq__(self, value: Any) -> bool:
+        ...
+
+
 array = TypeVar("array", bound="Array")
-device = TypeVar("device")
+device = TypeVar("device", bound=Device)
 dtype = TypeVar("dtype", bound="DType")
 SupportsDLPack = TypeVar("SupportsDLPack")
 SupportsBufferProtocol = TypeVar("SupportsBufferProtocol")

src/array_api_stubs/_draft/array_object.py

@@ -5,7 +5,7 @@
 from typing import TYPE_CHECKING, Protocol, TypeVar
 from enum import Enum
 from .data_types import DType
-from ._types import device as Device
+from ._types import Device
 
 if TYPE_CHECKING:
     from ._types import (
@@ -36,7 +36,7 @@ def dtype(self) -> DType:
         ...
 
     @property
-    def device(self) -> "Device":  # type: ignore[type-var]
+    def device(self) -> Device:
         """
         Hardware device the array data resides on.