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: ...
2323def is_dask_namespace (xp : ModuleType , / ) -> bool : ...
2424def is_jax_namespace (xp : ModuleType , / ) -> bool : ...
2525def is_numpy_namespace (xp : ModuleType , / ) -> bool : ...
26- def is_pydata_sparse_namespace (xp : ModuleType , / ) -> bool : ...
2726def is_torch_namespace (xp : ModuleType , / ) -> bool : ...
2827def is_jax_array (x : object , / ) -> bool : ...
28+ def is_pydata_sparse_namespace (xp : ModuleType , / ) -> bool : ...
2929def is_writeable_array (x : object , / ) -> bool : ...
3030def size (x : Array , / ) -> int | None : ...
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments