ohtddlmZddlmZddlmZmZddlmZm Z ddl Z dddd d Z dddd d Z dd Zdd Zy)) annotations)Array) _result_type_real_numeric_dtypes)OptionalTupleNFaxiskeepdimsc |jtvr tdtjt j t 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_arrayxr r s ^/var/www/html/engine/venv/lib/python3.12/site-packages/numpy/array_api/_searching_functions.pyrr F  ww**HII ::bjj188$!RS TTc |jtvr tdtjt j t 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 rrrrrc`tdtj|jDS)z~ Array API compatible wrapper for :py:func:`np.nonzero `. See its docstring for more information. c3FK|]}tj|yw)N)rr).0is r znonzero..'s=1A=s!)tuplernonzeror)rs rr"r"!s" = 188(<= ==rct|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) conditionx1x2s rr%r%*sV288$  & &r2 .FB ::bhhy//BIIF GGr)rrr z Optional[int]r boolreturnr)rrr*zTuple[Array, ...])r&rr'rr(rr*r) __future__r _array_objectr_dtypesrrtypingrr numpyrrrr"r%rrr1s9" 7"26U26U> Hr