Skip to content

Commit f1fb8d6

Browse files
committed
revert vendor changes
1 parent dc40a5e commit f1fb8d6

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

src/array_api_extra/_lib/_utils/_compat.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ def is_cupy_namespace(xp: ModuleType, /) -> bool: ...
2323
def is_dask_namespace(xp: ModuleType, /) -> bool: ...
2424
def is_jax_namespace(xp: ModuleType, /) -> bool: ...
2525
def is_numpy_namespace(xp: ModuleType, /) -> bool: ...
26-
def is_pydata_sparse_namespace(xp: ModuleType, /) -> bool: ...
2726
def is_torch_namespace(xp: ModuleType, /) -> bool: ...
2827
def is_jax_array(x: object, /) -> bool: ...
28+
def is_pydata_sparse_namespace(xp: ModuleType, /) -> bool: ...
2929
def is_writeable_array(x: object, /) -> bool: ...
3030
def size(x: Array, /) -> int | None: ...

vendor_tests/test_vendor.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ def test_vendor_compat():
77
array_namespace,
88
device,
99
is_cupy_namespace,
10-
is_dask_namespace,
1110
is_jax_array,
1211
is_jax_namespace,
1312
is_pydata_sparse_namespace,
@@ -19,13 +18,12 @@ def test_vendor_compat():
1918
x = xp.asarray([1, 2, 3])
2019
assert array_namespace(x) is xp
2120
device(x)
22-
assert not is_cupy_namespace(xp)
23-
assert not is_dask_namespace(xp)
2421
assert not is_jax_array(x)
22+
assert is_writeable_array(x)
23+
assert not is_cupy_namespace(xp)
2524
assert not is_jax_namespace(xp)
2625
assert not is_pydata_sparse_namespace(xp)
2726
assert not is_torch_namespace(xp)
28-
assert is_writeable_array(x)
2927
assert size(x) == 3
3028

3129

0 commit comments

Comments
 (0)