diff --git a/array_api_compat/common/_helpers.py b/array_api_compat/common/_helpers.py index 4d808e4b..434e7d87 100644 --- a/array_api_compat/common/_helpers.py +++ b/array_api_compat/common/_helpers.py @@ -776,7 +776,7 @@ def to_device(x: Array, device: Device, /, *, stream: Optional[Union[int, Any]] device : Hardware device the array data resides on. """ - if is_numpy_array(x) or is_ndonnx_array(x): + if is_numpy_array(x): if stream is not None: raise ValueError("The stream argument to to_device() is not supported") if device == 'cpu':