♻️ refactor _Call11Bool dtype parameter to use _DTypeLikeBool

Author: guan404ming

src/numpy-stubs/_typing/_ufunc.pyi

@@ -14,7 +14,7 @@ from typing_extensions import Never, TypeAliasType, TypeVar, Unpack
 
 import numpy as np
 from numpy import _CastingKind, _OrderKACF  # noqa: ICN003
-from numpy._typing import _NestedSequence
+from numpy._typing import _DTypeLikeBool, _NestedSequence
 
 from ._array_like import ArrayLike, NDArray, _ArrayLikeBool_co, _ArrayLikeInt_co
 from ._dtype_like import DTypeLike, _DTypeLike
@@ -187,7 +187,7 @@ class _Call11Bool(Protocol):
         /,
         out: None = None,
         *,
-        dtype: DTypeLike | None = None,
+        dtype: _DTypeLikeBool | None = None,
         **kwds: Unpack[_Kwargs2],
     ) -> np.bool: ...
     @overload  # (array-like, out: T) -> T
@@ -197,7 +197,7 @@ class _Call11Bool(Protocol):
         /,
         out: _Out1[_ArrayT],
         *,
-        dtype: DTypeLike | None = None,
+        dtype: _DTypeLikeBool | None = None,
         **kwds: Unpack[_Kwargs2],
     ) -> _ArrayT: ...
     @overload  # (array) -> Array[bool] | bool
@@ -207,7 +207,7 @@ class _Call11Bool(Protocol):
         /,
         out: _Out1[_AnyArray] | None = None,
         *,
-        dtype: DTypeLike | None = None,
+        dtype: _DTypeLikeBool | None = None,
         **kwds: Unpack[_Kwargs2],
     ) -> NDArray[np.bool]: ...
     @overload  # (array-like) -> Array[bool] | bool
@@ -217,7 +217,7 @@ class _Call11Bool(Protocol):
         /,
         out: _Out1[_AnyArray] | None = None,
         *,
-        dtype: DTypeLike | None = None,
+        dtype: _DTypeLikeBool | None = None,
         **kwds: Unpack[_Kwargs2],
     ) -> NDArray[np.bool] | np.bool: ...
 

test/static/reject/ufuncs.pyi

@@ -1,6 +1,7 @@
 import numpy as np
 import numpy.typing as npt
 
+i8: np.int64
 AR_f8: npt.NDArray[np.float64]
 
 np.sin.nin + "foo"  # type: ignore[operator]  # pyright: ignore[reportOperatorIssue]
@@ -40,3 +41,8 @@ np.matmul.accumulate()  # type: ignore[call-arg]  # pyright: ignore[reportCallIs
 np.frexp.at()  # type: ignore[call-arg, misc]  # pyright: ignore[reportCallIssue]
 np.divmod.at()  # type: ignore[call-arg, misc]  # pyright: ignore[reportCallIssue]
 np.matmul.at()  # type: ignore[call-arg, misc]  # pyright: ignore[reportCallIssue]
+
+np.isnan(i8, dtype=int)  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType, reportCallIssue]
+np.isnat(i8, dtype=np.int64)  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType, reportCallIssue]
+np.isinf(i8, dtype=np.int64)  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType, reportCallIssue]
+np.isfinite(i8, dtype=np.int64)  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType, reportCallIssue]