clean up

Author: lucascolley

src/array_api_extra/__init__.py

@@ -1,6 +1,6 @@
 """Extra array functions built on top of the array API standard."""
 
-from ._delegators import pad
+from ._delegation import pad
 from ._lib._funcs import (
     at,
     atleast_nd,

src/array_api_extra/_delegators.py renamed to src/array_api_extra/_delegation.py

@@ -1,4 +1,6 @@
-"""Delegators to existing implementations for Public API Functions."""
+"""Delegation to existing implementations for Public API Functions."""
+
+from types import ModuleType
 
 from ._lib import _funcs
 from ._lib._utils._compat import (
@@ -8,7 +10,7 @@
     is_numpy_namespace,
     is_torch_namespace,
 )
-from ._lib._utils._typing import Array, ModuleType
+from ._lib._utils._typing import Array
 
 
 def pad(

src/array_api_extra/_lib/__init__.py

@@ -1 +1 @@
-"""Array-agnostic implementations for the public API."""
+"""Internals of array-api-extra."""

src/array_api_extra/_lib/_funcs.py

@@ -1,4 +1,4 @@
-"""Public API Functions."""
+"""Array-agnostic implementations for the public API."""
 
 # https://github.com/scikit-learn/scikit-learn/pull/27910#issuecomment-2568023972
 from __future__ import annotations

src/array_api_extra/_lib/_utils/_helpers.py

@@ -3,8 +3,10 @@
 # https://github.com/scikit-learn/scikit-learn/pull/27910#issuecomment-2568023972
 from __future__ import annotations
 
+from types import ModuleType
+
 from . import _compat
-from ._typing import Array, ModuleType
+from ._typing import Array
 
 __all__ = ["in1d", "mean"]
 

src/array_api_extra/_lib/_utils/_typing.py

@@ -1,11 +1,10 @@
 """Static typing helpers."""
 
-from types import ModuleType
 from typing import Any
 
 # 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]
 Index = Any  # type: ignore[no-any-explicit]
 
-__all__ = ["Array", "Device", "Index", "ModuleType"]
+__all__ = ["Array", "Device", "Index"]

tests/test_at.py

@@ -13,7 +13,7 @@
 )
 
 from array_api_extra import at
-from array_api_extra._lib._typing import Array
+from array_api_extra._lib._utils._typing import Array
 
 all_libraries = (
     "array_api_strict",