|
1 | 1 | from array_api_compat import ( # noqa: F401 |
2 | 2 | is_numpy_array, is_cupy_array, is_torch_array, |
3 | 3 | is_dask_array, is_jax_array, is_pydata_sparse_array, |
| 4 | + is_ndonnx_array, |
4 | 5 | is_numpy_namespace, is_cupy_namespace, is_torch_namespace, |
5 | 6 | is_dask_namespace, is_jax_namespace, is_pydata_sparse_namespace, |
6 | | - is_array_api_strict_namespace, |
| 7 | + is_array_api_strict_namespace, is_ndonnx_namespace, |
7 | 8 | ) |
8 | 9 |
|
9 | 10 | from array_api_compat import device, is_array_api_obj, is_writeable_array, to_device |
|
22 | 23 | 'dask.array': 'is_dask_array', |
23 | 24 | 'jax.numpy': 'is_jax_array', |
24 | 25 | 'sparse': 'is_pydata_sparse_array', |
| 26 | + 'ndonnx': 'is_ndonnx_array', |
25 | 27 | } |
26 | 28 |
|
27 | 29 | is_namespace_functions = { |
|
32 | 34 | 'jax.numpy': 'is_jax_namespace', |
33 | 35 | 'sparse': 'is_pydata_sparse_namespace', |
34 | 36 | 'array_api_strict': 'is_array_api_strict_namespace', |
| 37 | + 'ndonnx': 'is_ndonnx_namespace', |
35 | 38 | } |
36 | 39 |
|
37 | 40 |
|
|
0 commit comments