o ԭj@s|ddlmZddlmZddlmZmZddlmZm Z ddl Z ddddddZ ddddddZ dddZdddZdS)) annotations)Array) _result_type_real_numeric_dtypes)OptionalTupleNFaxiskeepdimsxrr Optional[int]r boolreturnc C0|jtvr tdtttj|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 ra/var/www/html/CropPilot/venv/lib/python3.10/site-packages/numpy/array_api/_searching_functions.pyr  rc Cr)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 ) rrrrrrrargminrrrrrrrrTuple[Array, ...]cCstddt|jDS)z~ Array API compatible wrapper for :py:func:`np.nonzero `. See its docstring for more information. css|]}t|VqdS)N)rr).0irrr 'sznonzero..)tuplernonzeror)r rrrr"!sr" conditionx1x2cCs8t|j|jt||\}}tt|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%rrrr'*sr')r rr r r rrr)r rrr)r#rr$rr%rrr) __future__r _array_objectr_dtypesrrtypingrrnumpyrrrr"r'rrrrs