o
    TiS                     @   s0   d dl mZ G dd deZG dd deZdS )    )Enumc                   @   s.   e Zd ZdZ	 dZ	 dZ	 dZ	 dZ	 dZdS )SchedulingResultr                  N)	__name__
__module____qualname__SuccessEngineSequenceLimitExceededBatchSequenceLimitExceededBatchTokenLimitExceededKVCacheLimitExceededSequenceTokenLimitExceeded r   r   [/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/inference/v2/scheduling_utils.pyr   	   s    r   c                       s2   e Zd ZU eed< 	 deddf fddZ  ZS )SchedulingErrorresultreturnNc                    s   || _ t d|  d S )Nz$Batch scheduling failed with result )r   super__init__)selfr   	__class__r   r   r   4   s   zSchedulingError.__init__)r	   r
   r   r   __annotations__r   __classcell__r   r   r   r   r   -   s   
 r   N)enumr   r   RuntimeErrorr   r   r   r   r   <module>   s   $