Merge branch 'array-api-implementation' of https://github.com/amalia-k510/fast-array-utils into array-api-implementation

Author: amalia-k510

pyproject.toml

@@ -66,7 +66,7 @@ envs.docs.dependency-groups = [ "doc" ]
 envs.docs.scripts.build = "sphinx-build -M html docs docs/_build"
 envs.docs.scripts.clean = "git clean -fdX docs"
 envs.docs.scripts.open = "python -m webbrowser -t docs/_build/html/index.html"
-envs.hatch-test.default-args = [  ]
+envs.hatch-test.default-args = []
 envs.hatch-test.dependency-groups = [ "test-min" ]
 # TODO: remove scipy once https://github.com/pypa/hatch/pull/2127 is released
 envs.hatch-test.extra-dependencies = [ "ipykernel", "ipycytoscape", "scipy" ]
@@ -91,7 +91,7 @@ metadata.hooks.docstring-description = {}
 metadata.hooks.fancy-pypi-readme.content-type = "text/x-rst"
 metadata.hooks.fancy-pypi-readme.fragments = [ { path = "README.rst", start-after = ".. begin" } ]
 version.source = "vcs"
-version.raw-options = { local_scheme = "no-local-version" } # be able to publish dev version
+version.raw-options = { local_scheme = "no-local-version" }  # be able to publish dev version
 
 [tool.uv]
 override-dependencies = [ "sphinx>=9.0.1" ]
@@ -132,7 +132,7 @@ lint.per-file-ignores."typings/**/*.pyi" = [ "A002", "F403", "F405", "N801" ]
 lint.allowed-confusables = [ "×", "’" ]
 lint.flake8-bugbear.extend-immutable-calls = [ "testing.fast_array_utils.Flags" ]
 lint.flake8-copyright.notice-rgx = "SPDX-License-Identifier: MPL-2\\.0"
-lint.flake8-type-checking.exempt-modules = [  ]
+lint.flake8-type-checking.exempt-modules = []
 lint.flake8-type-checking.strict = true
 lint.isort.known-first-party = [ "fast_array_utils" ]
 lint.isort.lines-after-imports = 2

src/fast_array_utils/stats/_generic_ops.py

@@ -4,6 +4,7 @@
 from functools import singledispatch
 from typing import TYPE_CHECKING, cast, get_args
 
+import array_api_compat
 import numpy as np
 
 from .. import types
@@ -72,8 +73,6 @@ def _generic_op_array_api(
     """Handle arrays with native array API support."""
     del keep_cupy_as_array
 
-    import array_api_compat
-
     xp = array_api_compat.array_namespace(x)
     return getattr(xp, op)(x, axis=axis, **_dtype_kw(dtype, op))
 

src/fast_array_utils/stats/_is_constant.py

@@ -29,7 +29,6 @@ def is_constant_(
 @is_constant_.register(np.ndarray | types.CupyArray | types.HasArrayNamespace)
 def _is_constant_ndarray(a: NDArray[Any] | types.CupyArray, /, *, axis: Literal[0, 1] | None = None) -> bool | NDArray[np.bool] | types.CupyArray:
     # Should eventually support nd, not now.
-
     match axis:
         case None:
             return bool((a == a.reshape(-1)[0]).all())

src/fast_array_utils/stats/_power.py

@@ -4,6 +4,7 @@
 from functools import singledispatch
 from typing import TYPE_CHECKING
 
+import array_api_compat
 import numpy as np
 
 from .. import types
@@ -39,7 +40,6 @@ def _power_numpy(x: np.ndarray, n: int, /, dtype: DTypeLike | None = None) -> np
 
 @_power.register(types.HasArrayNamespace)
 def _power_array_api(x: types.HasArrayNamespace, n: int, /, dtype: DTypeLike | None = None) -> types.HasArrayNamespace:
-    import array_api_compat
 
     xp = array_api_compat.array_namespace(x)
     return xp.pow(x, n) if dtype is None else xp.pow(xp.astype(x, dtype), n)