remove numpy.array_api

xadupre · xadupre · commit f07bb4b2491e · 2025-02-17T16:54:58.000+01:00
diff --git a/_unittests/ut_array_api/test_hypothesis_array_api.py b/_unittests/ut_array_api/test_hypothesis_array_api.py
@@ -1,5 +1,4 @@
 import unittest
-import warnings
 from os import getenv
 from functools import reduce
 import packaging.version as pv
@@ -45,12 +44,7 @@ class TestHypothesisArraysApis(ExtTestCase):
 
     @classmethod
     def setUpClass(cls):
-        try:
-            import array_api_strict as xp
-        except ImportError:
-            with warnings.catch_warnings():
-                warnings.simplefilter("ignore")
-                from numpy import array_api as xp
+        import array_api_strict as xp
 
         api_version = getenv(
             "ARRAY_API_TESTS_VERSION",
diff --git a/onnx_array_api/array_api/_onnx_common.py b/onnx_array_api/array_api/_onnx_common.py
@@ -1,16 +1,7 @@
 from typing import Any, Optional
-import warnings
 import numpy as np
 from onnx import TensorProto
-
-try:
-    import array_api_strict
-
-    Array = type(array_api_strict.ones((1,)))
-except ImportError:
-    with warnings.catch_warnings():
-        warnings.simplefilter("ignore")
-        from numpy.array_api._array_object import Array
+import array_api_strict
 
 from ..npx.npx_types import (
     DType,
@@ -36,6 +27,9 @@
 )
 
 
+Array = type(array_api_strict.ones((1,)))
+
+
 # These functions with no specific code do not have to be
 # implemented. They are automatically added in
 # :mod:`onnx_array_api.array_api`. It needs
