o
    $i                     @   sH   d dl mZ d dlmZ d dlmZ d dlmZ eG dd deZdS )    )defaultdict)Dict)Stopper)	PublicAPIc                   @   s8   e Zd ZdZdefddZdedefddZd	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                 C   s   || _ tdd | _d S )Nc                   S      dS )Nr    r	   r	   r	   _/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/tune/stopper/maximum_iteration.py<lambda>   s    z2MaximumIterationStopper.__init__.<locals>.<lambda>)	_max_iterr   _iter)selfr   r	   r	   r
   __init__   s   z MaximumIterationStopper.__init__trial_idresultc                 C   s"   | j |  d7  < | j | | jkS )N   )r   r   )r   r   r   r	   r	   r
   __call__   s   z MaximumIterationStopper.__call__c                 C   r   )NFr	   )r   r	   r	   r
   stop_all   s   z MaximumIterationStopper.stop_allN)
__name__
__module____qualname____doc__intr   strr   r   r   r	   r	   r	   r
   r      s
    r   N)	collectionsr   typingr   ray.tune.stopper.stopperr   ray.util.annotationsr   r   r	   r	   r	   r
   <module>   s    