a ;e@sddlmZddlmZddlmZmZddlmZm Z ddl Z dddd d d d d d dZ dddd d d d d ddZ d ddddZd d d d dddZdS)) annotations)Array) _result_type_real_numeric_dtypes)OptionalTupleNFaxiskeepdimsrz Optional[int]bool)xr r returnc Cs0|jtvrtdtttj|j||dS)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 rX/var/www/sunrise/env/lib/python3.9/site-packages/numpy/array_api/_searching_functions.pyr s rc Cs0|jtvrtdtttj|j||dS)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 ) rrrrrrrargminrrrrrrs rzTuple[Array, ...])r rcCstddt|jDS)z~ Array API compatible wrapper for :py:func:`np.nonzero `. See its docstring for more information. css|]}t|VqdS)N)rr).0irrr 'znonzero..)tuplernonzeror)r rrrr!sr) conditionx1x2rcCs8t|j|jt||\}}tt|j|j|jS)zz Array API compatible wrapper for :py:func:`np.where `. See its docstring for more information. )rrrZ_normalize_two_argsrrwherer)r r!r"rrrr#*sr#) __future__rZ _array_objectrZ_dtypesrrtypingrrnumpyrrrrr#rrrrs