xbarray 0.0.1a1__py3-none-any.whl

array_api_typing/typing_compat/__init__.py

@@ -0,0 +1,12 @@
+from ._array_typing import Array as ArrayAPIArray, Device as ArrayAPIDevice, DType as ArrayAPIDType, PyCapsule, SupportsDLPack, SetIndex, GetIndex
+from ._api_typing import ArrayAPINamespace
+
+__all__ = [
+    "ArrayAPIArray",
+    "ArrayAPIDevice",
+    "ArrayAPIDType",
+    "SetIndex",
+    "GetIndex",
+    "SupportsDLPack",
+    "ArrayAPINamespace",
+]

array_api_typing/typing_compat/_api_typing.py

@@ -0,0 +1,27 @@
+from typing import Protocol, TypeVar, Optional, Any, Tuple, Union, Type
+from typing_extensions import deprecated
+from abc import abstractmethod
+from array_api_typing.typing_2024_12._api_typing import ArrayAPINamespace as ArrayAPI202412Namespace, _NAMESPACE_C, _NAMESPACE_ARRAY, _NAMESPACE_DEVICE, _NAMESPACE_DTYPE
+
+class ArrayAPINamespace(ArrayAPI202412Namespace[_NAMESPACE_ARRAY, _NAMESPACE_DEVICE, _NAMESPACE_DTYPE], Protocol[_NAMESPACE_ARRAY, _NAMESPACE_DEVICE, _NAMESPACE_DTYPE]):
+    @abstractmethod
+    def device(self, x : _NAMESPACE_ARRAY, /) -> _NAMESPACE_DEVICE:
+        pass
+
+    @abstractmethod
+    def to_device(
+        self,
+        x: _NAMESPACE_ARRAY,
+        device: _NAMESPACE_DEVICE,
+        /,
+        *,
+        stream : Optional[Union[int, Any]] = None,
+    ) -> _NAMESPACE_ARRAY:
+        pass
+
+    @abstractmethod
+    def size(
+        self,
+        x: _NAMESPACE_ARRAY
+    ) -> Optional[int]:
+        pass

array_api_typing/typing_compat/_array_typing.py

@@ -0,0 +1,36 @@
+from typing import Protocol, TypeVar, Optional, Any, Tuple, Union, Type
+from typing_extensions import deprecated
+from array_api_typing.typing_2024_12._array_typing import SetIndex, GetIndex, Array as ArrayAPI202412Array, Device, DType, PyCapsule, SupportsDLPack
+
+__all__ = [
+    "SetIndex",
+    "GetIndex",
+    "Array",
+    "PyCapsule",
+    "SupportsDLPack",
+    "DType",
+    "Device",
+]
+
+class Array(ArrayAPI202412Array, Protocol):
+    @property
+    @deprecated(
+        "Some array libraries either do not have the device attribute or include it with an incompatible API. Use array_api_namespace.device() instead.",
+    )
+    def device(self) -> Device:
+        pass
+    
+    @deprecated(
+        "some array libraries do not have the `to_device` method. Use array_api_namespace.to_device() instead.",
+    )
+    def to_device(
+        self, device: Device, /, *, stream: Optional[Union[int, Any]] = None
+    ):
+        pass
+    
+    @property
+    @deprecated(
+        "PyTorch defines size in an incompatible way. It also fixes dask.arrays behaviour which returns nan for unknown sizes, whereas the standard requires None.",
+    )
+    def size(self) -> Optional[int]:
+        pass

array_api_typing/typing_extra/__init__.py

@@ -0,0 +1,12 @@
+from array_api_typing.typing_compat import ArrayAPIArray, ArrayAPIDevice, ArrayAPIDType, SupportsDLPack, SetIndex, GetIndex
+from ._api_typing import ArrayAPINamespace
+
+__all__ = [
+    "ArrayAPIArray",
+    "ArrayAPIDevice",
+    "ArrayAPIDType",
+    "SetIndex",
+    "GetIndex",
+    "SupportsDLPack",
+    "ArrayAPINamespace"
+]