o
    ʩZh                     @  s|   d dl mZ ddlmZ ddlmZmZ d dlmZm	Z	 d dl
ZddddddZddddddZdddZdddZdS )    )annotations   )Array)_result_type_real_numeric_dtypes)OptionalTupleNFaxiskeepdimsxr   r
   Optional[int]r   boolreturnc            	   C  0   | j tvr	tdtttj| j||dS )z|
    Array API compatible wrapper for :py:func:`np.argmax <numpy.argmax>`.

    See its docstring for more information.
    z.Only real numeric dtypes are allowed in argmaxr	   )	dtyper   	TypeErrorr   _newnpasarrayargmax_arrayr   r
   r    r   [/var/www/html/lang_env/lib/python3.10/site-packages/numpy/array_api/_searching_functions.pyr         
r   c            	   C  r   )z|
    Array API compatible wrapper for :py:func:`np.argmin <numpy.argmin>`.

    See its docstring for more information.
    z.Only real numeric dtypes are allowed in argminr	   )	r   r   r   r   r   r   r   argminr   r   r   r   r   r      r   r   Tuple[Array, ...]c                C  s   t dd t| jD S )z~
    Array API compatible wrapper for :py:func:`np.nonzero <numpy.nonzero>`.

    See its docstring for more information.
    c                 s  s    | ]}t |V  qd S )N)r   r   ).0ir   r   r   	<genexpr>'   s    znonzero.<locals>.<genexpr>)tupler   nonzeror   )r   r   r   r   r"   !   s   r"   	conditionx1x2c                C  s8   t |j|j t||\}}tt| j|j|jS )zz
    Array API compatible wrapper for :py:func:`np.where <numpy.where>`.

    See its docstring for more information.
    )r   r   r   Z_normalize_two_argsr   r   wherer   )r#   r$   r%   r   r   r   r&   *   s   r&   )r   r   r
   r   r   r   r   r   )r   r   r   r   )r#   r   r$   r   r%   r   r   r   )
__future__r   Z_array_objectr   Z_dtypesr   r   typingr   r   numpyr   r   r   r"   r&   r   r   r   r   <module>   s    
	