xbarray 0.0.1a3__tar.gz → 0.0.1a5__tar.gz

{xbarray-0.0.1a3/xbarray.egg-info → xbarray-0.0.1a5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xbarray
-Version: 0.0.1a3
+Version: 0.0.1a5
 Summary: Cross-backend Python array library based on the Array API Standard.
 Requires-Python: >=3.10
 License-File: LICENSE

{xbarray-0.0.1a3 → xbarray-0.0.1a5}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "xbarray"
 description = "Cross-backend Python array library based on the Array API Standard."
-version = "0.0.1a3"
+version = "0.0.1a5"
 requires-python = ">= 3.10"
 dependencies = [
     "typing_extensions>=4.5",

{xbarray-0.0.1a3 → xbarray-0.0.1a5}/xbarray/_src/implementations/_common/implementations.py

@@ -1,5 +1,6 @@
 from typing import Any, Union, Callable
 from array_api_typing.typing_compat import ArrayAPINamespace as CompatNamespace, ArrayAPIArray as CompatArray, ArrayAPIDType as CompatDType
+import array_api_compat
 
 __all__ = [
     "get_abbreviate_array_function",
@@ -19,7 +20,7 @@ def get_abbreviate_array_function(
         Or, if some dimensions are the same, abbreviates to a smaller array (but with the same number of dimensions).
         """
         abbr_array = array
-        idx = backend.zeros(1, dtype=default_integer_dtype, device=backend.device(abbr_array))
+        idx = backend.zeros(1, dtype=default_integer_dtype, device=array_api_compat.device(abbr_array))
         for dim_i in range(len(array.shape)):
             first_elem = backend.take(abbr_array, idx, axis=dim_i)
             if backend.all(abbr_array == first_elem):

{xbarray-0.0.1a3 → xbarray-0.0.1a5}/xbarray/_src/implementations/jax/__init__.py

@@ -1,7 +1,5 @@
 import jax.numpy
 
-simplified_name = "jax"
-
 if hasattr(jax.numpy, "__array_api_version__"):
     compat_module = jax.numpy
     from jax.numpy import *
@@ -11,6 +9,8 @@ else:
 
 from array_api_compat.common._helpers import *
 
+simplified_name = "jax"
+
 # Import and bind all functions from array_api_extra before exposing them
 import array_api_extra
 from functools import partial

{xbarray-0.0.1a3 → xbarray-0.0.1a5}/xbarray/_src/implementations/jax/_extra.py

@@ -57,7 +57,6 @@ def to_numpy(
     return np.asarray(data)
 
 def to_dlpack(
-    self,
     data: ARRAY_TYPE,
     /,
 ) -> SupportsDLPack:

{xbarray-0.0.1a3 → xbarray-0.0.1a5}/xbarray/_src/implementations/numpy/__init__.py

@@ -1,4 +1,5 @@
 from array_api_compat.numpy import *
+from array_api_compat.common._helpers import *
 
 simplified_name = "numpy"
 

{xbarray-0.0.1a3 → xbarray-0.0.1a5}/xbarray/_src/implementations/numpy/_extra.py

@@ -48,7 +48,6 @@ def to_numpy(
     return data
 
 def to_dlpack(
-    self,
     data: ARRAY_TYPE,
     /,
 ) -> SupportsDLPack:

{xbarray-0.0.1a3 → xbarray-0.0.1a5}/xbarray/_src/implementations/pytorch/__init__.py

@@ -1,4 +1,5 @@
 from array_api_compat.torch import *
+from array_api_compat.common._helpers import *
 
 simplified_name = "pytorch"
 

{xbarray-0.0.1a3 → xbarray-0.0.1a5}/xbarray/_src/implementations/pytorch/_extra.py

@@ -65,7 +65,6 @@ def to_numpy(
     return data.cpu().numpy()
 
 def to_dlpack(
-    self,
     data: ARRAY_TYPE,
     /,
 ) -> SupportsDLPack:

{xbarray-0.0.1a3 → xbarray-0.0.1a5/xbarray.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xbarray
-Version: 0.0.1a3
+Version: 0.0.1a5
 Summary: Cross-backend Python array library based on the Array API Standard.
 Requires-Python: >=3.10
 License-File: LICENSE