o
    `Û·i  ã                   @   s0   d dl mZ d dlmZ eG dd„ deƒƒZdS )é    )ÚOldAPIStack)ÚPolynomialSchedulec                       s    e Zd ZdZ‡ fdd„Z‡  ZS )ÚLinearSchedulezÅLinear interpolation between `initial_p` and `final_p`.

    Uses `PolynomialSchedule` with power=1.0.


    The formula is:
    value = `final_p` + (`initial_p` - `final_p`) * (1 - `t`/t_max)
    c                    s   t ƒ jdddi|¤Ž dS )z&Initializes a LinearSchedule instance.Úpowerg      ð?N© )ÚsuperÚ__init__)ÚselfÚkwargs©Ú	__class__r   ú_/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/rllib/utils/schedules/linear_schedule.pyr      s   zLinearSchedule.__init__)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   Ú__classcell__r   r   r   r   r      s    	r   N)Úray.rllib.utils.annotationsr   Ú-ray.rllib.utils.schedules.polynomial_scheduler   r   r   r   r   r   Ú<module>   s    