xbarray 0.0.1a3__py3-none-any.whl → 0.0.1a4__py3-none-any.whl

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/_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/_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/_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.dist-info → xbarray-0.0.1a4.dist-info}/METADATA

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

{xbarray-0.0.1a3.dist-info → xbarray-0.0.1a4.dist-info}/RECORD

@@ -16,16 +16,16 @@ xbarray/__init__.py,sha256=k4Ipp7IoODqHWZ-eeBhcU7Ch8FudF8rag6KbeIurrgY,45
 xbarray/jax.py,sha256=3DHAFsWvfurCI0H6loqx4LtmBov9KbIxeHGmwLKrdh4,344
 xbarray/numpy.py,sha256=8thyuJZlm0MJXGx6YWlgxV9ASG-In9jn-pwUFFk7iB4,358
 xbarray/pytorch.py,sha256=NEKlF_QoiOQl0T-aP_CkUxJ3HgFJEm0hmLy9v5I7qAo,372
-xbarray/_src/implementations/_common/implementations.py,sha256=FyZaQZcVdSnH6CUA1LVbUJAK-EAlkz2fDfOOuVyku6Y,2655
-xbarray/_src/implementations/jax/__init__.py,sha256=pkQASOOePuTel32zaSmsNIc9FqL_8Ix0P4NDXMqhupE,827
+xbarray/_src/implementations/_common/implementations.py,sha256=ba6fyoHMrawrrjTULSqt073-aMw7abqFWh5YnzeOHMU,2688
+xbarray/_src/implementations/jax/__init__.py,sha256=fZMtXujTLhfHofWJzeMtsOuNhUYSZzCEj06RulT4RqQ,827
 xbarray/_src/implementations/jax/_extra.py,sha256=2akggZTP2xIXe5AdVs4rNk8yCJOpTJX8rlv94SY4HJA,2525
 xbarray/_src/implementations/jax/_typing.py,sha256=U9BUxHNEjFB0LHF1KMrFLbh6E5kVvpAF8bZUbLNf25E,278
 xbarray/_src/implementations/jax/random.py,sha256=k6vuNLTOSUClMKI6My0-Xekkeu6hJrUyNib1TziuGag,3312
-xbarray/_src/implementations/numpy/__init__.py,sha256=s_tC8UT0ULs9tpZ5JFiEIOnv5sRAVBZFM5Wh90T44Wc,640
+xbarray/_src/implementations/numpy/__init__.py,sha256=UcUDdf7xDp_oLw5O73G_n15qYzPd5XQsq4cnZoaX9y0,687
 xbarray/_src/implementations/numpy/_extra.py,sha256=p7WkiCWEl14A4W6lwo2TthKalsuAe7DuRhtQxJWf7ng,2003
 xbarray/_src/implementations/numpy/_typing.py,sha256=pgjLLAipwFsIk0QdgrAA4PEvjF_ugHbzfSTbLpA__6o,241
 xbarray/_src/implementations/numpy/random.py,sha256=C1z2-pMcDMRa7nKhKNTuY5LtTlu9nXvDpH1wY8mdIug,2630
-xbarray/_src/implementations/pytorch/__init__.py,sha256=CIXK76WnheuEkVjdU_efCgLxBx3ABhcTdbaImRzmkNM,643
+xbarray/_src/implementations/pytorch/__init__.py,sha256=ZCGmZpbv4NaCWU_08G-ipEpI1xMr7QLXzEhwrhD7fI8,690
 xbarray/_src/implementations/pytorch/_extra.py,sha256=XBYZEYcia3JT7Uk2PT-DyiC-_9J_Rjtg3vQB04spXc8,2888
 xbarray/_src/implementations/pytorch/_typing.py,sha256=qSnNZD3IpgSoOTl3TBRBv2ifSAHOw0aR9uyzfV5KYVw,204
 xbarray/_src/implementations/pytorch/random.py,sha256=p4ZDsTc--z5XslVzel-iEVDtoKizrv39_1tfR4-469o,2657
@@ -34,8 +34,8 @@ xbarray/_src/serialization/serialization_map.py,sha256=05zfqip8qPCt-LgM_cx_zLc4y
 xbarray/base/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
 xbarray/base/base.py,sha256=V1yAXu2wUu6q50okX7JQ8onkGhHIQ0lkPkYOolwc45A,5928
 xbarray/cls_impl/cls_base.py,sha256=MUvJeMm4UVW4jXwfVP02GiCzqftrAND-JYvThge4PUw,312
-xbarray-0.0.1a3.dist-info/licenses/LICENSE,sha256=6P0HCOancSfch0dNycuDIe8_qwS0Id97Ih_8hjJ2PFI,1067
-xbarray-0.0.1a3.dist-info/METADATA,sha256=GQAkLlOrxDmTlTR7Tsex2UQ3mjvcVKqOxKlCy1bxBGE,436
-xbarray-0.0.1a3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xbarray-0.0.1a3.dist-info/top_level.txt,sha256=VriXuFyU48Du4HQMzROSArhwqB6EZYY0n0mipgUqB9A,25
-xbarray-0.0.1a3.dist-info/RECORD,,
+xbarray-0.0.1a4.dist-info/licenses/LICENSE,sha256=6P0HCOancSfch0dNycuDIe8_qwS0Id97Ih_8hjJ2PFI,1067
+xbarray-0.0.1a4.dist-info/METADATA,sha256=Ij9lDaD2N4y8zUe5wU_QrXxtAJpDS1zmZmhU8RUli2I,436
+xbarray-0.0.1a4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xbarray-0.0.1a4.dist-info/top_level.txt,sha256=VriXuFyU48Du4HQMzROSArhwqB6EZYY0n0mipgUqB9A,25
+xbarray-0.0.1a4.dist-info/RECORD,,