TYP: work around a quantum-entangled mypy issue, somehow

Author: jorenham

numpy/_core/fromnumeric.pyi

@@ -3,14 +3,15 @@ from collections.abc import Sequence
 from typing import (
     Any,
     Literal,
-    NoReturn,
     Protocol,
     SupportsIndex,
     TypeAlias,
     TypeVar,
     overload,
     type_check_only,
 )
+
+from _typeshed import Incomplete
 from typing_extensions import Never, deprecated
 
 import numpy as np
@@ -556,9 +557,6 @@ def ravel(
 @overload
 def ravel(a: ArrayLike, order: _OrderKACF = "C") -> np.ndarray[tuple[int], np.dtype[Any]]: ...
 
-@overload
-def nonzero(a: np.generic | np.ndarray[tuple[()], Any]) -> NoReturn: ...
-@overload
 def nonzero(a: _ArrayLike[Any]) -> tuple[NDArray[intp], ...]: ...
 
 # this prevents `Any` from being returned with Pyright
@@ -818,7 +816,7 @@ def all(
     keepdims: _BoolLike_co | _NoValueType = ...,
     *,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> np.bool | NDArray[np.bool]: ...
+) -> Incomplete: ...
 @overload
 def all(
     a: ArrayLike,
@@ -855,7 +853,7 @@ def any(
     keepdims: _BoolLike_co | _NoValueType = ...,
     *,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> np.bool | NDArray[np.bool]: ...
+) -> Incomplete: ...
 @overload
 def any(
     a: ArrayLike,
@@ -1448,10 +1446,10 @@ def mean(
     keepdims: Literal[False] | _NoValueType = ...,
     *,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> complexfloating[Any, Any]: ...
+) -> complexfloating[Any]: ...
 @overload
 def mean(
-    a: _ArrayLikeTD64_co,
+    a: _ArrayLike[np.timedelta64],
     axis: None = ...,
     dtype: None = ...,
     out: None = ...,
@@ -1462,23 +1460,33 @@ def mean(
 @overload
 def mean(
     a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
-    axis: _ShapeLike | None = ...,
-    dtype: None = ...,
-    out: None = ...,
+    axis: _ShapeLike | None,
+    dtype: DTypeLike,
+    out: _ArrayT,
     keepdims: bool | _NoValueType = ...,
     *,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> Any: ...
+) -> _ArrayT: ...
+@overload
+def mean(
+    a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
+    axis: _ShapeLike | None = ...,
+    dtype: DTypeLike | None = ...,
+    *,
+    out: _ArrayT,
+    keepdims: bool | _NoValueType = ...,
+    where: _ArrayLikeBool_co | _NoValueType = ...,
+) -> _ArrayT: ...
 @overload
 def mean(
     a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
     axis: None,
     dtype: _DTypeLike[_SCT],
     out: None = ...,
-    keepdims: bool | _NoValueType = ...,
+    keepdims: Literal[False] | _NoValueType = ...,
     *,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> _SCT | NDArray[_SCT]: ...
+) -> _SCT: ...
 @overload
 def mean(
     a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
@@ -1492,43 +1500,43 @@ def mean(
 @overload
 def mean(
     a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
-    axis: None = ...,
-    *,
+    axis: _ShapeLike | None,
     dtype: _DTypeLike[_SCT],
-    out: None = ...,
-    keepdims: bool | _NoValueType = ...,
+    out: None,
+    keepdims: Literal[True, 1],
+    *,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> _SCT | NDArray[_SCT]: ...
+) -> NDArray[_SCT]: ...
 @overload
 def mean(
     a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
-    axis: _ShapeLike | None = ...,
-    dtype: DTypeLike = ...,
+    axis: _ShapeLike | None,
+    dtype: _DTypeLike[_SCT],
     out: None = ...,
-    keepdims: bool | _NoValueType = ...,
     *,
+    keepdims: bool | _NoValueType = ...,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> Any: ...
+) -> _SCT | NDArray[_SCT]: ...
 @overload
 def mean(
     a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
-    axis: _ShapeLike | None,
-    dtype: DTypeLike,
-    out: _ArrayT,
-    keepdims: bool | _NoValueType = ...,
+    axis: _ShapeLike | None = ...,
     *,
+    dtype: _DTypeLike[_SCT],
+    out: None = ...,
+    keepdims: bool | _NoValueType = ...,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> _ArrayT: ...
+) -> _SCT | NDArray[_SCT]: ...
 @overload
 def mean(
     a: _ArrayLikeComplex_co | _ArrayLikeObject_co,
     axis: _ShapeLike | None = ...,
-    dtype: DTypeLike = ...,
-    *,
-    out: _ArrayT,
+    dtype: DTypeLike | None = ...,
+    out: None = ...,
     keepdims: bool | _NoValueType = ...,
+    *,
     where: _ArrayLikeBool_co | _NoValueType = ...,
-) -> _ArrayT: ...
+) -> Incomplete: ...
 
 @overload
 def std(

numpy/typing/tests/data/reveal/fromnumeric.pyi

@@ -1,6 +1,6 @@
 """Tests for :mod:`_core.fromnumeric`."""
 
-from typing import Any, Literal as L, NoReturn
+from typing import Any, Literal as L
 
 import numpy as np
 import numpy.typing as npt
@@ -126,11 +126,8 @@ assert_type(np.ravel(f), np.ndarray[tuple[int], np.dtype[np.float64 | np.int_ |
 assert_type(np.ravel(AR_b), np.ndarray[tuple[int], np.dtype[np.bool]])
 assert_type(np.ravel(AR_f4), np.ndarray[tuple[int], np.dtype[np.float32]])
 
-assert_type(np.nonzero(b), NoReturn)
-assert_type(np.nonzero(f4), NoReturn)
 assert_type(np.nonzero(AR_b), tuple[npt.NDArray[np.intp], ...])
 assert_type(np.nonzero(AR_f4), tuple[npt.NDArray[np.intp], ...])
-assert_type(np.nonzero(AR_0d), NoReturn)
 assert_type(np.nonzero(AR_1d), tuple[npt.NDArray[np.intp], ...])
 assert_type(np.nonzero(AR_nd), tuple[npt.NDArray[np.intp], ...])