
    &`i                     b    d dl mZ d dlmZ d dlmZ d dlmZ e G d de                      ZdS )    )defaultdict)Dict)Stopper)	PublicAPIc                   4    e Zd ZdZdefdZdedefdZd Z	dS )	MaximumIterationStopperzStop trials after reaching a maximum number of iterations

    Args:
        max_iter: Number of iterations before stopping a trial.
    max_iterc                 >    || _         t          d           | _        d S )Nc                      dS )Nr    r       v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/stopper/maximum_iteration.py<lambda>z2MaximumIterationStopper.__init__.<locals>.<lambda>   s     r   )	_max_iterr   _iter)selfr	   s     r   __init__z MaximumIterationStopper.__init__   s    ! ++


r   trial_idresultc                 X    | j         |xx         dz  cc<   | j         |         | j        k    S )N   )r   r   )r   r   r   s      r   __call__z MaximumIterationStopper.__call__   s4    
8!z(#t~55r   c                     dS )NFr   )r   s    r   stop_allz MaximumIterationStopper.stop_all   s    ur   N)
__name__
__module____qualname____doc__intr   strr   r   r   r   r   r   r   r      sj         , , , , ,6 6d 6 6 6 6    r   r   N)	collectionsr   typingr   ray.tune.stopper.stopperr   ray.util.annotationsr   r   r   r   r   <module>r%      s    # # # # # #       , , , , , , * * * * * *     g     r   