From a7dfb2e1a8794376cac3e981e6f82f98403b2729 Mon Sep 17 00:00:00 2001 From: crusaderky Date: Thu, 20 Feb 2025 15:56:09 +0000 Subject: [PATCH] Revert change to ndonnx .to_device --- array_api_compat/common/_helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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':