clean up

Author: lucascolley

src/array_api_extra/__init__.py

@@ -1,4 +1,4 @@
-from __future__ import annotations
+from __future__ import annotations  # https://github.com/pylint-dev/pylint/pull/9990
 
 from ._funcs import atleast_nd, cov, create_diagonal, expand_dims, kron, setdiff1d, sinc
 

src/array_api_extra/_lib/_typing.py

@@ -6,16 +6,17 @@
 
 if typing.TYPE_CHECKING:
     from typing_extensions import override
+
+    # To be changed to a Protocol later (see data-apis/array-api#589)
+    Array = Any  # type: ignore[no-any-explicit]
+    Device = Any  # type: ignore[no-any-explicit]
 else:
 
     def no_op_decorator(f):  # pyright: ignore[reportUnreachable]
         return f
 
     override = no_op_decorator
 
-__all__ = ["Array", "Device", "ModuleType", "override"]
-
-
-# To be changed to a Protocol later (see data-apis/array-api#589)
-Array = Any  # type: ignore[no-any-explicit]
-Device = Any  # type: ignore[no-any-explicit]
+__all__ = ["ModuleType", "override"]
+if typing.TYPE_CHECKING:
+    __all__ += ["Array", "Device"]