|
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