NotificationsYou must be signed in to change notification settings
Fork10.9k
Star29.5k

Commitd1ffd4f

committed

TYP: np.argmin changes

1 parent00f2733 commitd1ffd4fCopy full SHA for d1ffd4f

File tree

2 files changed

+14

-15

lines changed

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

2 files changed

+14

-15

lines changed

`‎numpy/_core/fromnumeric.pyi`

Lines changed: 9 additions & 15 deletions

Original file line number	Diff line number	Diff line change
`@@ -111,6 +111,7 @@ _NumberOrObjectT = TypeVar("_NumberOrObjectT", bound=np.number \| np.object_)`
`111`	`111`	`_ArrayT=TypeVar("_ArrayT",bound=np.ndarray[Any,Any])`
`112`	`112`	`_ShapeT=TypeVar("_ShapeT",bound=tuple[int, ...])`
`113`	`113`	`_ShapeT_co=TypeVar("_ShapeT_co",bound=tuple[int, ...],covariant=True)`
	`114`	`+_BoolOrIntArrayT=TypeVar("_BoolOrIntArrayT",bound=NDArray[np.integer\|np.bool])`
`114`	`115`
`115`	`116`	`@type_check_only`
`116`	`117`	`class_SupportsShape(Protocol[_ShapeT_co]):`
`@@ -418,18 +419,18 @@ def argmax(`
`418`	`419`	`defargmax(`
`419`	`420`	`a:ArrayLike,`
`420`	`421`	`axis:SupportsIndex\|None,`
`421`		`-out:_ArrayT,`
	`422`	`+out:_BoolOrIntArrayT,`
`422`	`423`	`*,`
`423`	`424`	`keepdims:bool= ...,`
`424`		`-)->_ArrayT: ...`
	`425`	`+)->_BoolOrIntArrayT: ...`
`425`	`426`	`@overload`
`426`	`427`	`defargmax(`
`427`	`428`	`a:ArrayLike,`
`428`	`429`	`axis:SupportsIndex\|None= ...,`
`429`	`430`	`*,`
`430`		`-out:_ArrayT,`
	`431`	`+out:_BoolOrIntArrayT,`
`431`	`432`	`keepdims:bool= ...,`
`432`		`-)->_ArrayT: ...`
	`433`	`+)->_BoolOrIntArrayT: ...`
`433`	`434`
`434`	`435`	`@overload`
`435`	`436`	`defargmin(`
`@@ -451,26 +452,19 @@ def argmin(`
`451`	`452`	`defargmin(`
`452`	`453`	`a:ArrayLike,`
`453`	`454`	`axis:SupportsIndex\|None,`
`454`		`-out:_ArrayT,`
	`455`	`+out:_BoolOrIntArrayT,`
`455`	`456`	`*,`
`456`	`457`	`keepdims:bool= ...,`
`457`		`-)->_ArrayT: ...`
	`458`	`+)->_BoolOrIntArrayT: ...`
`458`	`459`	`@overload`
`459`	`460`	`defargmin(`
`460`	`461`	`a:ArrayLike,`
`461`	`462`	`axis:SupportsIndex\|None= ...,`
`462`	`463`	`*,`
`463`		`-out:_ArrayT,`
	`464`	`+out:_BoolOrIntArrayT,`
`464`	`465`	`keepdims:bool= ...,`
`465`		`-)->_ArrayT: ...`
	`466`	`+)->_BoolOrIntArrayT: ...`
`466`	`467`
`467`		`-@overload`
`468`		`-defsearchsorted(`
`469`		`-a:ArrayLike,`
`470`		`-v:_ScalarLike_co,`
`471`		`-side:_SortSide= ...,`
`472`		`-sorter:_ArrayLikeInt_co\|None= ...,# 1D int array`
`473`		`-)->intp: ...`
`474`	`468`	`@overload`
`475`	`469`	`defsearchsorted(`
`476`	`470`	`a:ArrayLike,`

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

Lines changed: 5 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -97,6 +97,11 @@ assert_type(np.argmin(AR_b, axis=0), Any)`
`97`	`97`	`assert_type(np.argmin(AR_f4,axis=0),Any)`
`98`	`98`	`assert_type(np.argmin(AR_f4,out=AR_subclass),NDArraySubclass)`
`99`	`99`
	`100`	`+OUT_I=np.empty((),dtype=np.intp)`
	`101`	`+OUT_F=np.empty((),dtype=np.float64)`
	`102`	`+assert_type(np.argmin(AR_f4,out=OUT_I),npt.NDArray[np.intp])`
	`103`	`+assert_type(np.argmax(AR_f4,out=OUT_I),npt.NDArray[np.intp])`
	`104`	`+`
`100`	`105`	`assert_type(np.searchsorted(AR_b[0],0),np.intp)`
`101`	`106`	`assert_type(np.searchsorted(AR_f4[0],0),np.intp)`
`102`	`107`	`assert_type(np.searchsorted(AR_b[0], [0]),npt.NDArray[np.intp])`

0 commit comments

Comments

(0)

Movatterモバイル変換

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Uh oh!

Commitd1ffd4f

File tree

2 files changed

2 files changed

`‎numpy/_core/fromnumeric.pyi`

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

0 commit comments