Skip to content

Commit ddb03e6

Browse files
committed
Run tests on ndonnx
1 parent b87e0aa commit ddb03e6

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

tests/_helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
wrapped_libraries = ["numpy", "cupy", "torch", "dask.array"]
77
all_libraries = wrapped_libraries + [
8-
"array_api_strict", "jax.numpy", "sparse"
8+
"array_api_strict", "jax.numpy", "sparse", "ndonnx"
99
]
1010

1111
# `sparse` added array API support as of Python 3.10.

tests/test_common.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
from array_api_compat import ( # noqa: F401
22
is_numpy_array, is_cupy_array, is_torch_array,
33
is_dask_array, is_jax_array, is_pydata_sparse_array,
4+
is_ndonnx_array,
45
is_numpy_namespace, is_cupy_namespace, is_torch_namespace,
56
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,
78
)
89

910
from array_api_compat import device, is_array_api_obj, is_writeable_array, to_device
@@ -22,6 +23,7 @@
2223
'dask.array': 'is_dask_array',
2324
'jax.numpy': 'is_jax_array',
2425
'sparse': 'is_pydata_sparse_array',
26+
'ndonnx': 'is_ndonnx_array',
2527
}
2628

2729
is_namespace_functions = {
@@ -32,6 +34,7 @@
3234
'jax.numpy': 'is_jax_namespace',
3335
'sparse': 'is_pydata_sparse_namespace',
3436
'array_api_strict': 'is_array_api_strict_namespace',
37+
'ndonnx': 'is_ndonnx_namespace',
3538
}
3639

3740

0 commit comments

Comments
 (0)