mipy issues fix

Author: amalia-k510

src/fast_array_utils/conv/__init__.py

@@ -73,4 +73,4 @@ def to_dense(
     Dense form of ``x``
 
     """
-    return to_dense_(x, order=order, to_cpu_memory=to_cpu_memory)
+    return to_dense_(x, order=order, to_cpu_memory=to_cpu_memory)  # type: ignore[no-any-return]

src/fast_array_utils/stats/__init__.py

@@ -77,7 +77,7 @@ def is_constant(
     from ._is_constant import is_constant_
 
     validate_axis(x.ndim, axis)
-    return is_constant_(x, axis=axis)
+    return is_constant_(x, axis=axis)  # type: ignore[no-any-return]
 
 
 # TODO(flying-sheep): support CSDataset (TODO)
@@ -226,7 +226,7 @@ def _generic_op(
         assert dtype is None or op in get_args(DtypeOps), f"`dtype` is not supported for operation {op!r}"
 
         validate_axis(x.ndim, axis)
-        return generic_op(x, op, axis=axis, keep_cupy_as_array=keep_cupy_as_array, dtype=dtype)
+        return generic_op(x, op, axis=axis, keep_cupy_as_array=keep_cupy_as_array, dtype=dtype)  # type: ignore[no-any-return]
 
     _generic_op.__name__ = op
     return cast("StatFunNoDtype | StatFunDtype", _generic_op)

src/fast_array_utils/stats/_generic_ops.py

@@ -24,7 +24,7 @@
 
 @singledispatch
 def generic_op(
-    x: CpuArray | GpuArray | DiskArray | types.DaskArray,
+    x: CpuArray | GpuArray | DiskArray | types.DaskArray | types.HasArrayNamespace,
     /,
     op: Ops,
     *,

src/fast_array_utils/stats/_power.py

@@ -10,7 +10,9 @@
 
 
 if TYPE_CHECKING:
-    from numpy.typing import Any, DTypeLike
+    from typing import Any
+
+    from numpy.typing import DTypeLike
 
     from fast_array_utils.typing import CpuArray, GpuArray
 
@@ -21,7 +23,7 @@
 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[return-value]
+    return _power(x, n, dtype=dtype)  # type: ignore[no-any-return]
 
 
 @singledispatch