array

xadupre · xadupre · commit 202562c0a031 · 2024-08-30T14:25:24.000+02:00
"
diff --git a/onnx_array_api/array_api/_onnx_common.py b/onnx_array_api/array_api/_onnx_common.py
@@ -3,9 +3,12 @@
 import numpy as np
 from onnx import TensorProto
 
-with warnings.catch_warnings():
-    warnings.simplefilter("ignore")
-    from numpy.array_api._array_object import Array
+try:
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore")
+        from numpy.array_api._array_object import Array
+except ImportError:
+    Array = None
 from ..npx.npx_types import (
     DType,
     ElemType,
@@ -107,7 +110,7 @@ def asarray(
             v = TEagerTensor(np.array(a))
     elif isinstance(a, np.ndarray):
         v = TEagerTensor(a)
-    elif isinstance(a, Array):
+    elif Array and isinstance(a, Array):
         v = TEagerTensor(np.asarray(a))
     else:
         raise RuntimeError(f"Unexpected type {type(a)} for the first input.")
