Refine count_nonzero typing stub

mtsokol · mtsokol · commit 1ceb3dcd7fa5 · 2025-04-22T10:42:42.000Z
diff --git a/numpy/_core/numeric.pyi b/numpy/_core/numeric.pyi
@@ -416,7 +416,7 @@ def full_like(
 
 #
 @overload
-def count_nonzero(a: ArrayLike, axis: None = None, *, keepdims: L[False] = False) -> int: ...
+def count_nonzero(a: ArrayLike, axis: None = None, *, keepdims: L[False] = False) -> np.intp: ...
 @overload
 def count_nonzero(a: _ScalarLike_co, axis: _ShapeLike | None = None, *, keepdims: L[True]) -> np.intp: ...
 @overload
diff --git a/numpy/typing/tests/data/reveal/numeric.pyi b/numpy/typing/tests/data/reveal/numeric.pyi
@@ -25,9 +25,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), npt.NDArray[np.intp])
 assert_type(np.count_nonzero(AR_i8, axis=0), Any)
 

Original file line number	Diff line number	Diff line change
`@@ -416,7 +416,7 @@ def full_like(`
`416`	`416`
`417`	`417`	`#`
`418`	`418`	`@overload`
`419`		`-def count_nonzero(a: ArrayLike, axis: None = None, *, keepdims: L[False] = False) -> int: ...`
	`419`	`+def count_nonzero(a: ArrayLike, axis: None = None, *, keepdims: L[False] = False) -> np.intp: ...`
`420`	`420`	`@overload`
`421`	`421`	`def count_nonzero(a: _ScalarLike_co, axis: _ShapeLike \| None = None, *, keepdims: L[True]) -> np.intp: ...`
`422`	`422`	`@overload`