xbarray 0.0.1a5__py3-none-any.whl → 0.0.1a7__py3-none-any.whl

array_api_typing/typing_2024_12/_api_return_typing.py

@@ -98,6 +98,6 @@ class Info(Protocol):
         ...
 
     def dtypes(
-        self, *, device: Optional[Device], kind: Optional[Union[str, Tuple[str, ...]]]
-    ) -> DataTypes:
+        self, *, device: Optional[Device] = None, kind: Optional[Union[str, Tuple[str, ...]]] = None
+    ) -> DataTypes: # In the original signature there's no default parameters for `device` and `kind`, but we add them since they exist in `array_api_compat` modules
         ...

xbarray/_src/implementations/jax/__init__.py

@@ -3,9 +3,11 @@ import jax.numpy
 if hasattr(jax.numpy, "__array_api_version__"):
     compat_module = jax.numpy
     from jax.numpy import *
+    from jax.numpy import __array_api_version__, __array_namespace_info__
 else:
     import jax.experimental.array_api as compat_module
     from jax.experimental.array_api import *
+    from jax.experimental.array_api import __array_api_version__, __array_namespace_info__
 
 from array_api_compat.common._helpers import *
 
@@ -18,7 +20,7 @@ for api_name in dir(array_api_extra):
     if api_name.startswith('_'):
         continue
 
-    if api_name == 'at':
+    if api_name in ['at', 'broadcast_shapes']:
         globals()[api_name] = getattr(array_api_extra, api_name)
     else:
         globals()[api_name] = partial(

xbarray/_src/implementations/numpy/__init__.py

@@ -1,4 +1,5 @@
 from array_api_compat.numpy import *
+from array_api_compat.numpy import __array_api_version__, __array_namespace_info__
 from array_api_compat.common._helpers import *
 
 simplified_name = "numpy"
@@ -11,7 +12,7 @@ for api_name in dir(array_api_extra):
     if api_name.startswith('_'):
         continue
 
-    if api_name == 'at':
+    if api_name in ['at', 'broadcast_shapes']:
         globals()[api_name] = getattr(array_api_extra, api_name)
     else:
         globals()[api_name] = partial(

xbarray/_src/implementations/pytorch/__init__.py

@@ -1,4 +1,5 @@
 from array_api_compat.torch import *
+from array_api_compat.torch import __array_api_version__, __array_namespace_info__
 from array_api_compat.common._helpers import *
 
 simplified_name = "pytorch"
@@ -12,7 +13,7 @@ for api_name in dir(array_api_extra):
     if api_name.startswith('_'):
         continue
 
-    if api_name == 'at':
+    if api_name in ['at', 'broadcast_shapes']:
         globals()[api_name] = getattr(array_api_extra, api_name)
     else:
         globals()[api_name] = partial(

xbarray/jax.py

@@ -2,10 +2,16 @@ from xbarray.cls_impl.cls_base import ComputeBackendImplCls
 from ._src.implementations import jax as jax_impl
 
 class JaxComputeBackend(metaclass=ComputeBackendImplCls):
-    pass
+    ARRAY_TYPE = jax_impl.ARRAY_TYPE
+    DTYPE_TYPE = jax_impl.DTYPE_TYPE
+    DEVICE_TYPE = jax_impl.DEVICE_TYPE
+    RNG_TYPE = jax_impl.RNG_TYPE
 
 for name in dir(jax_impl):
-    if not name.startswith('_'):
+    if not name.startswith('_') or name in [
+        '__array_namespace_info__',
+        '__array_api_version__',
+    ]:
         setattr(JaxComputeBackend, name, getattr(jax_impl, name))
 
 __all__ = [

xbarray/numpy.py

@@ -2,10 +2,16 @@ from xbarray.cls_impl.cls_base import ComputeBackendImplCls
 from ._src.implementations import numpy as numpy_impl
 
 class NumpyComputeBackend(metaclass=ComputeBackendImplCls):
-    pass
+    ARRAY_TYPE = numpy_impl.ARRAY_TYPE
+    DTYPE_TYPE = numpy_impl.DTYPE_TYPE
+    DEVICE_TYPE = numpy_impl.DEVICE_TYPE
+    RNG_TYPE = numpy_impl.RNG_TYPE
 
 for name in dir(numpy_impl):
-    if not name.startswith('_'):
+    if not name.startswith('_') or name in [
+        '__array_namespace_info__',
+        '__array_api_version__',
+    ]:
         setattr(NumpyComputeBackend, name, getattr(numpy_impl, name))
 
 __all__ = [

xbarray/pytorch.py

@@ -2,10 +2,16 @@ from xbarray.cls_impl.cls_base import ComputeBackendImplCls
 from ._src.implementations import pytorch as pytorch_impl
 
 class PytorchComputeBackend(metaclass=ComputeBackendImplCls):
-    pass
+    ARRAY_TYPE = pytorch_impl.ARRAY_TYPE
+    DTYPE_TYPE = pytorch_impl.DTYPE_TYPE
+    DEVICE_TYPE = pytorch_impl.DEVICE_TYPE
+    RNG_TYPE = pytorch_impl.RNG_TYPE
 
 for name in dir(pytorch_impl):
-    if not name.startswith('_'):
+    if not name.startswith('_') or name in [
+        '__array_namespace_info__',
+        '__array_api_version__',
+    ]:
         setattr(PytorchComputeBackend, name, getattr(pytorch_impl, name))
 
 __all__ = [

{xbarray-0.0.1a5.dist-info → xbarray-0.0.1a7.dist-info}/METADATA

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

{xbarray-0.0.1a5.dist-info → xbarray-0.0.1a7.dist-info}/RECORD

@@ -3,7 +3,7 @@ array_api_typing/typing_2024_12/__init__.py,sha256=JuD2yojHl3eQI74U75tFhSX5rSy32
 array_api_typing/typing_2024_12/_api_constant.py,sha256=tahdzdi7vAZ3UnM8oDcUqvx2FFsMTAyUbuwgarU6K9U,676
 array_api_typing/typing_2024_12/_api_fft_typing.py,sha256=i48S-9trw3s72xiCJmeCNGlJLwwOVq0kJC4lD9se_co,38641
 array_api_typing/typing_2024_12/_api_linalg_typing.py,sha256=Dt5fDTwYb-GAmP7duajwPqH3zaJmwvxDEOOzdl5X0LE,48950
-array_api_typing/typing_2024_12/_api_return_typing.py,sha256=7TxSBRv0H0uQ9VnixbyuEXJQnl1uESo102RsAloI9SA,2302
+array_api_typing/typing_2024_12/_api_return_typing.py,sha256=rIygF4PAr7G1PK3cVOwQNrev_kGCuMcfIJ9i8L0u1tI,2462
 array_api_typing/typing_2024_12/_api_typing.py,sha256=rFuF7E6b2RrijzQBrZx6_dYEn9Yslq8FqI31L1Ccrf4,294255
 array_api_typing/typing_2024_12/_array_typing.py,sha256=GMqs5LgshujRIh7i0mGtZy4Vt56FSXqBmNlBGidFRnc,57403
 array_api_typing/typing_compat/__init__.py,sha256=JuD2yojHl3eQI74U75tFhSX5rSy32TR5nKyi714ZfX4,347
@@ -13,19 +13,19 @@ array_api_typing/typing_extra/__init__.py,sha256=YfdhD-Sfk3SCfI9lHmA-PbVLzms1OFF
 array_api_typing/typing_extra/_api_typing.py,sha256=Jj_E61r35EgecWmBvAzpASV4qub5aQI_O4aL-ngEvQ8,23028
 array_api_typing/typing_extra/_at.py,sha256=S7_YjOwR3a8olZWgwpLDFEfnekRufRqrtfiMLwrWjgo,2202
 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/jax.py,sha256=nEdufdTHGXsQSYBbCUwgLikR7saAoZ0zL1a4vrs4I2w,569
+xbarray/numpy.py,sha256=1EaT-RVrJEsrgXE-JmbrKfH6nTVLi9sjQ3D81rWhd4Y,591
+xbarray/pytorch.py,sha256=4j1UOD6wC29zIAFOR6s0tV2vY2bqKCiiJfua_k_jHKs,613
 xbarray/_src/implementations/_common/implementations.py,sha256=ba6fyoHMrawrrjTULSqt073-aMw7abqFWh5YnzeOHMU,2688
-xbarray/_src/implementations/jax/__init__.py,sha256=fZMtXujTLhfHofWJzeMtsOuNhUYSZzCEj06RulT4RqQ,827
+xbarray/_src/implementations/jax/__init__.py,sha256=vOtCkwe98vNI1qQOu0Xv9OIaVucaJp6bEI9_mzicBH0,1014
 xbarray/_src/implementations/jax/_extra.py,sha256=QmPL7Syp-6y2NUbMYsTlI8uER0Gh5xjBbrNeyTIGZsc,2515
 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=UcUDdf7xDp_oLw5O73G_n15qYzPd5XQsq4cnZoaX9y0,687
+xbarray/_src/implementations/numpy/__init__.py,sha256=hJEcFR86pb1G72MFS1X15qwI_-BuPxy5SS5PXB0alQI,792
 xbarray/_src/implementations/numpy/_extra.py,sha256=JJs0_HNsnEkBPlXsomRjX5Jh254iXuw_o_3ZXK5c83s,1993
 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=ZCGmZpbv4NaCWU_08G-ipEpI1xMr7QLXzEhwrhD7fI8,690
+xbarray/_src/implementations/pytorch/__init__.py,sha256=s2mF3jWI356oedPZevKYARCFhlXH0hqS9OgBzxMSgHQ,795
 xbarray/_src/implementations/pytorch/_extra.py,sha256=HwytMuHZ0iyYKmYCj6QaTUKukMRWb47pme6xJ07LLjY,2878
 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.1a5.dist-info/licenses/LICENSE,sha256=6P0HCOancSfch0dNycuDIe8_qwS0Id97Ih_8hjJ2PFI,1067
-xbarray-0.0.1a5.dist-info/METADATA,sha256=MDImlQt3Kh9anmVussaPDDswLzAwb85TeIn1lW0-v9A,436
-xbarray-0.0.1a5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xbarray-0.0.1a5.dist-info/top_level.txt,sha256=VriXuFyU48Du4HQMzROSArhwqB6EZYY0n0mipgUqB9A,25
-xbarray-0.0.1a5.dist-info/RECORD,,
+xbarray-0.0.1a7.dist-info/licenses/LICENSE,sha256=6P0HCOancSfch0dNycuDIe8_qwS0Id97Ih_8hjJ2PFI,1067
+xbarray-0.0.1a7.dist-info/METADATA,sha256=MIiS1RQ5y8CdBbW_h7UvEMwO_QyaRnX_5BhI8CzV8Jc,436
+xbarray-0.0.1a7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xbarray-0.0.1a7.dist-info/top_level.txt,sha256=VriXuFyU48Du4HQMzROSArhwqB6EZYY0n0mipgUqB9A,25
+xbarray-0.0.1a7.dist-info/RECORD,,