small fix for coverage

data-apis · Nov 23, 2024 · b8b72b6 · b8b72b6
1 parent 3eb5f15
commit b8b72b6
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 8 deletions.
diff --git a/src/array_api_extra/_compat.py b/src/array_api_extra/_compat.py
@@ -4,9 +4,9 @@
 
 import inspect
 import sys
-from typing import TYPE_CHECKING
+import typing
 
-if TYPE_CHECKING:
+if typing.TYPE_CHECKING:
     from ._typing import Array, Device
 
 __all__ = ["device"]

diff --git a/src/array_api_extra/_funcs.py b/src/array_api_extra/_funcs.py
@@ -1,9 +1,9 @@
 from __future__ import annotations  # https://github.com/pylint-dev/pylint/pull/9990
 
+import typing
 import warnings
-from typing import TYPE_CHECKING
 
-if TYPE_CHECKING:
+if typing.TYPE_CHECKING:
     from ._typing import Array, ModuleType
 
 from . import _utils

diff --git a/src/array_api_extra/_utils.py b/src/array_api_extra/_utils.py
@@ -1,8 +1,8 @@
 from __future__ import annotations  # https://github.com/pylint-dev/pylint/pull/9990
 
-from typing import TYPE_CHECKING
+import typing
 
-if TYPE_CHECKING:
+if typing.TYPE_CHECKING:
     from ._typing import Array, ModuleType
 
 from . import _compat

diff --git a/tests/test_funcs.py b/tests/test_funcs.py
@@ -1,8 +1,9 @@
 from __future__ import annotations
 
 import contextlib
+import typing
 import warnings
-from typing import TYPE_CHECKING, Any
+from typing import Any
 
 # array-api-strict#6
 import array_api_strict as xp  # type: ignore[import-untyped]
@@ -12,7 +13,7 @@
 
 from array_api_extra import atleast_nd, cov, create_diagonal, expand_dims, kron, sinc
 
-if TYPE_CHECKING:
+if typing.TYPE_CHECKING:
     Array = Any  # To be changed to a Protocol later (see array-api#589)