Merge pull request #777 from numpy/np24/size-axis

jorenham · web-flow · commit 116a9c2d6434 · 2025-12-24T14:48:32.000+01:00
diff --git a/src/numpy-stubs/_core/fromnumeric.pyi b/src/numpy-stubs/_core/fromnumeric.pyi
@@ -1232,7 +1232,7 @@ def cumulative_prod(
 
 #
 def ndim(a: ArrayLike) -> int: ...
-def size(a: ArrayLike, axis: int | None = None) -> int: ...
+def size(a: ArrayLike, axis: int | tuple[int, ...] | None = None) -> int: ...
 
 #
 @overload

Original file line number	Diff line number	Diff line change
`@@ -1232,7 +1232,7 @@ def cumulative_prod(`
`1232`	`1232`
`1233`	`1233`	`#`
`1234`	`1234`	`def ndim(a: ArrayLike) -> int: ...`
`1235`		`-def size(a: ArrayLike, axis: int \| None = None) -> int: ...`
	`1235`	`+def size(a: ArrayLike, axis: int \| tuple[int, ...] \| None = None) -> int: ...`
`1236`	`1236`
`1237`	`1237`	`#`
`1238`	`1238`	`@overload`