xbarray 0.0.1a6__tar.gz → 0.0.1a7__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xbarray
-Version: 0.0.1a6
+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.1a6 → 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.1a6 → 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.1a6"
+version = "0.0.1a7"
 requires-python = ">= 3.10"
 dependencies = [
     "typing_extensions>=4.5",

{xbarray-0.0.1a6 → 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 *
 

{xbarray-0.0.1a6 → 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"

{xbarray-0.0.1a6 → 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"

{xbarray-0.0.1a6 → xbarray-0.0.1a7}/xbarray/jax.py

@@ -8,7 +8,10 @@ class JaxComputeBackend(metaclass=ComputeBackendImplCls):
     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.1a6 → xbarray-0.0.1a7}/xbarray/numpy.py

@@ -8,7 +8,10 @@ class NumpyComputeBackend(metaclass=ComputeBackendImplCls):
     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.1a6 → xbarray-0.0.1a7}/xbarray/pytorch.py

@@ -8,7 +8,10 @@ class PytorchComputeBackend(metaclass=ComputeBackendImplCls):
     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.1a6 → xbarray-0.0.1a7/xbarray.egg-info}/PKG-INFO

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