ignore comments update and mypy test

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
@@ -179,6 +179,10 @@ implicit_reexport = true
 explicit_package_bases = true
 mypy_path = [ "$MYPY_CONFIG_FILE_DIR/typings", "$MYPY_CONFIG_FILE_DIR/src" ]
 
+[[tool.mypy.overrides]]
+module = [ "jax", "jax.*" ]
+ignore_missing_imports = true
+
 [tool.pyright]
 stubPath = "./typings"
 reportPrivateUsage = false

src/fast_array_utils/conv/_to_dense.py

@@ -21,7 +21,7 @@
 # fallback’s arg0 type has to include types of registered functions
 @singledispatch
 def to_dense_(
-    x: CpuArray | GpuArray | DiskArray | types.DaskArray | types.sparray | types.spmatrix | types.CupySpMatrix,
+    x: CpuArray | GpuArray | DiskArray | types.DaskArray | types.sparray | types.spmatrix | types.CupySpMatrix | types.HasArrayNamespace,
     /,
     *,
     order: Literal["K", "A", "C", "F"] = "K",

src/fast_array_utils/stats/_generic_ops.py

@@ -56,7 +56,7 @@ def _generic_op_numpy(
 ) -> NDArray[Any] | np.number[Any]:
     del keep_cupy_as_array
     arr = getattr(np, op)(x, axis=axis, **_dtype_kw(dtype, op))
-    return arr.toarray() if isinstance(arr, types.CupyCOOMatrix) else arr
+    return arr.toarray() if isinstance(arr, types.CupyCOOMatrix) else arr  # type: ignore[return-value]
 
 
 @generic_op.register(types.HasArrayNamespace)

src/fast_array_utils/stats/_power.py

@@ -21,14 +21,14 @@
 def power[Arr: Array](x: Arr, n: int, /, dtype: DTypeLike | None = None) -> Arr:
     """Take array or matrix to a power."""
     # This wrapper is necessary because TypeVars can’t be used in `singledispatch` functions
-    return _power(x, n, dtype=dtype)  # type: ignore[no-any-return]
+    return _power(x, n, dtype=dtype)  # type: ignore[return-value]
 
 
 @singledispatch
 def _power(x: Array, n: int, /, dtype: DTypeLike | None = None) -> Array:
     if TYPE_CHECKING:
         assert not isinstance(x, types.DaskArray | types.CSBase | types.CupyCSMatrix)
-    return x**n if dtype is None else np.power(x, n, dtype=dtype)  # type: ignore[operator]
+    return x**n if dtype is None else np.power(x, n, dtype=dtype)  # type: ignore[arg-type]
 
 
 @_power.register(np.ndarray)
@@ -42,7 +42,7 @@ def _power_array_api(x: types.HasArrayNamespace, n: int, /, dtype: DTypeLike | N
     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)
+    return xp.pow(x, n) if dtype is None else xp.pow(xp.astype(x, dtype), n)  # type: ignore[no-any-return]
 
 
 @_power.register(types.CSBase | types.CupyCSMatrix)