bgvddlmZddlmZddlmZmZddlmZm Z ddl Z dddddZ dddddZ ddZddZdS)) annotations)Array) _result_type_real_numeric_dtypes)OptionalTupleNFaxiskeepdimsxrr Optional[int]r boolreturnc |jtvrtdtjt jt j|j||S)z| Array API compatible wrapper for :py:func:`np.argmax `. See its docstring for more information. z.Only real numeric dtypes are allowed in argmaxr ) dtyper TypeErrorr_newnpasarrayargmax_arrayr r r s w/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/array_api/_searching_functions.pyrr N  w***HIII :bj18$!R!R!RSS T TTc |jtvrtdtjt jt j|j||S)z| Array API compatible wrapper for :py:func:`np.argmin `. See its docstring for more information. z.Only real numeric dtypes are allowed in argminr ) rrrrrrrargminrrs rrrrrTuple[Array, ...]cbtdtj|jDS)z~ Array API compatible wrapper for :py:func:`np.nonzero `. See its docstring for more information. c3>K|]}tj|VdS)N)rr).0is r znonzero..'s*==1A======r)tuplernonzeror)r s rr&r&!s- == 18(<(<=== = ==r conditionx1x2ct|j|jtj||\}}tjt j|j|j|jS)zz Array API compatible wrapper for :py:func:`np.where `. See its docstring for more information. )rrr_normalize_two_argsrrwherer)r'r(r)s rr,r,*sS28$$$  &r2 . .FB :bhy/BIFF G GGr)r rr rr rrr)r rrr)r'rr(rr)rrr) __future__r _array_objectr_dtypesrrtypingrr numpyrrrr&r,rrr3s"""""" 77777777""""""""26UUUUUU26UUUUUU>>>> H H H H H Hr