small fix for coverage

Author: lucascolley

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"]

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

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)