iter

Author: lucascolley

vendor_tests/_array_api_compat_vendor.py

@@ -4,11 +4,11 @@
 from types import ModuleType
 from typing import Any
 
-from .array_api_compat import *  # noqa: F403
+from .array_api_compat import *  # type: ignore[import-not-found]  # noqa: F403
 from .array_api_compat import array_namespace as array_namespace_compat
 
 
 # Let unit tests check with `is` that we are picking up the function from this module
 # and not from the original array_api_compat module.
-def array_namespace(*xs: Any | complex | None, **kwargs) -> ModuleType:  # type: ignore[no-redef] # numpydoc ignore=GL08
+def array_namespace(*xs: Any | complex | None, **kwargs) -> ModuleType:  # pyrefly: ignore[unannotated-parameter] # numpydoc ignore=GL08
     return array_namespace_compat(*xs, **kwargs)

vendor_tests/test_vendor.py

@@ -5,7 +5,9 @@
 import array_api_strict as xp
 from numpy.testing import assert_array_equal
 
-from vendor_tests.array_api_compat.common._typing import Array
+from vendor_tests.array_api_compat.common._typing import (  # type: ignore[import-not-found]
+    Array,
+)
 
 
 def test_vendor_compat():
@@ -56,7 +58,7 @@ def test_vendor_compat():
 
 
 def test_vendor_extra():
-    from .array_api_extra import atleast_nd
+    from .array_api_extra import atleast_nd  # type: ignore[import-not-found]
 
     x = xp.asarray(1)
     x = cast(Array, x)
@@ -65,7 +67,9 @@ def test_vendor_extra():
 
 
 def test_vendor_extra_testing():
-    from .array_api_extra.testing import lazy_xp_function
+    from .array_api_extra.testing import (  # type: ignore[import-not-found]
+        lazy_xp_function,
+    )
 
     def f(x: Any) -> Any:
         return x
@@ -75,6 +79,8 @@ def f(x: Any) -> Any:
 
 def test_vendor_extra_uses_vendor_compat():
     from ._array_api_compat_vendor import array_namespace as n1
-    from .array_api_extra._lib._utils._compat import array_namespace as n2
+    from .array_api_extra._lib._utils._compat import (  # type: ignore[import-not-found]
+        array_namespace as n2,
+    )
 
     assert n1 is n2