Refine count_nonzero typing stub

mtsokol · mtsokol · commit 87b638955bd4 · 2025-04-02T22:01:22.000Z
diff --git a/numpy/_core/numeric.pyi b/numpy/_core/numeric.pyi
@@ -431,7 +431,7 @@ def count_nonzero(
     axis: None = ...,
     *,
     keepdims: L[False] = ...,
-) -> int: ...
+) -> intp: ...
 @overload
 def count_nonzero(
     a: ArrayLike,
diff --git a/numpy/typing/tests/data/reveal/numeric.pyi b/numpy/typing/tests/data/reveal/numeric.pyi
@@ -27,9 +27,9 @@ AR_O: npt.NDArray[np.object_]
 B: list[int]
 C: SubClass
 
-assert_type(np.count_nonzero(i8), int)
-assert_type(np.count_nonzero(AR_i8), int)
-assert_type(np.count_nonzero(B), int)
+assert_type(np.count_nonzero(i8), np.intp)
+assert_type(np.count_nonzero(AR_i8), np.intp)
+assert_type(np.count_nonzero(B), np.intp)
 assert_type(np.count_nonzero(AR_i8, keepdims=True), Any)
 assert_type(np.count_nonzero(AR_i8, axis=0), Any)
 
