xbarray 0.0.1a5__tar.gz → 0.0.1a7__tar.gz

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

@@ -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 → xbarray-0.0.1a7}/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-0.0.1a5 → xbarray-0.0.1a7}/pyproject.toml

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

{xbarray-0.0.1a5 → xbarray-0.0.1a7}/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-0.0.1a5 → xbarray-0.0.1a7}/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-0.0.1a5 → xbarray-0.0.1a7}/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-0.0.1a5 → xbarray-0.0.1a7}/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-0.0.1a5 → xbarray-0.0.1a7}/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-0.0.1a5 → xbarray-0.0.1a7}/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 → xbarray-0.0.1a7/xbarray.egg-info}/PKG-INFO

@@ -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