File tree Expand file tree Collapse file tree 2 files changed +3
-5
lines changed Expand file tree Collapse file tree 2 files changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -23,8 +23,8 @@ def is_cupy_namespace(xp: ModuleType, /) -> bool: ...
23
23
def is_dask_namespace (xp : ModuleType , / ) -> bool : ...
24
24
def is_jax_namespace (xp : ModuleType , / ) -> bool : ...
25
25
def is_numpy_namespace (xp : ModuleType , / ) -> bool : ...
26
- def is_pydata_sparse_namespace (xp : ModuleType , / ) -> bool : ...
27
26
def is_torch_namespace (xp : ModuleType , / ) -> bool : ...
28
27
def is_jax_array (x : object , / ) -> bool : ...
28
+ def is_pydata_sparse_namespace (xp : ModuleType , / ) -> bool : ...
29
29
def is_writeable_array (x : object , / ) -> bool : ...
30
30
def size (x : Array , / ) -> int | None : ...
Original file line number Diff line number Diff line change @@ -7,7 +7,6 @@ def test_vendor_compat():
7
7
array_namespace ,
8
8
device ,
9
9
is_cupy_namespace ,
10
- is_dask_namespace ,
11
10
is_jax_array ,
12
11
is_jax_namespace ,
13
12
is_pydata_sparse_namespace ,
@@ -19,13 +18,12 @@ def test_vendor_compat():
19
18
x = xp .asarray ([1 , 2 , 3 ])
20
19
assert array_namespace (x ) is xp
21
20
device (x )
22
- assert not is_cupy_namespace (xp )
23
- assert not is_dask_namespace (xp )
24
21
assert not is_jax_array (x )
22
+ assert is_writeable_array (x )
23
+ assert not is_cupy_namespace (xp )
25
24
assert not is_jax_namespace (xp )
26
25
assert not is_pydata_sparse_namespace (xp )
27
26
assert not is_torch_namespace (xp )
28
- assert is_writeable_array (x )
29
27
assert size (x ) == 3
30
28
31
29
You can’t perform that action at this time.
0 commit comments