lint

Author: crusaderky

pixi.lock

@@ -12,19 +12,17 @@
         is_jax_namespace,
         is_pydata_sparse_namespace,
         is_torch_namespace,
-        is_jax_namespace,
         is_writeable_array,
         size,
     )
 except ImportError:
     from array_api_compat import (  # pyright: ignore[reportMissingTypeStubs]
         array_namespace,
         device,
-        is_dask_namespace,
         is_cupy_namespace,
+        is_dask_namespace,
         is_jax_array,
         is_jax_namespace,
-        is_jax_namespace,
         is_pydata_sparse_namespace,
         is_torch_namespace,
         is_writeable_array,
@@ -40,7 +38,6 @@
     "is_jax_namespace",
     "is_pydata_sparse_namespace",
     "is_torch_namespace",
-    "is_jax_namespace",
     "is_writeable_array",
     "size",
 ]

src/array_api_extra/_lib/_compat.py

@@ -22,7 +22,6 @@ def is_cupy_namespace(x: object, /) -> bool: ...
 def is_dask_namespace(xp: ModuleType, /) -> bool: ...
 def is_jax_namespace(xp: ModuleType, /) -> bool: ...
 def is_jax_array(x: object, /) -> bool: ...
-def is_jax_namespace(x: object, /) -> bool: ...
 def is_pydata_sparse_namespace(x: object, /) -> bool: ...
 def is_torch_namespace(x: object, /) -> bool: ...
 def is_writeable_array(x: object, /) -> bool: ...

src/array_api_extra/_lib/_compat.pyi

@@ -12,22 +12,20 @@ def test_vendor_compat():
         is_jax_namespace,
         is_pydata_sparse_namespace,
         is_torch_namespace,
-        is_jax_namespace,
         is_writeable_array,
         size,
     )
 
     x = xp.asarray([1, 2, 3])
     assert array_namespace(x) is xp
     device(x)
+    assert not is_cupy_namespace(xp)
     assert not is_dask_namespace(xp)
-    assert not is_jax_namespace(xp)
     assert not is_jax_array(x)
-    assert is_writeable_array(x)
-    assert not is_cupy_namespace(xp)
     assert not is_jax_namespace(xp)
     assert not is_pydata_sparse_namespace(xp)
     assert not is_torch_namespace(xp)
+    assert is_writeable_array(x)
     assert size(x) == 3