o
    `Û·i5  ã                   @   s2  U d dl Z d dlmZ d dlmZ d dlmZmZ dZdZ	dZ
dZd	Zd
ZdZeed< dZdZeed< dZdZeed< dZdZeed< dZdZeed< dZdZdZdZdZdZdZdZ dZ!dZ"d Z#eed!< d"Z$e%ed#< d$Z&e%ed%< d&Z'd'Z(eeeeeeeeee e"ehZ)d(Z*d)e+fd*d+„Z,d)ee-e-f fd,d-„Z.dS ).é    N)ÚDict)Ú*RAY_WARN_BLOCKING_GET_INSIDE_ASYNC_ENV_VAR)Úenv_boolÚenv_set_by_userÚUNSUPPORTEDÚ
DEPRECATEDz.validate_storage_markerz checkpoint_manager_snapshot.json)zAWS Error SLOW_DOWNzAWS Error INTERNAL_FAILUREzAWS Error SERVICE_UNAVAILABLEzAWS Error NETWORK_CONNECTIONzAWS Error UNKNOWNÚ!RAY_TRAIN_HEALTH_CHECK_INTERVAL_Sg       @ÚDEFAULT_HEALTH_CHECK_INTERVAL_SÚ'RAY_TRAIN_WORKER_HEALTH_CHECK_TIMEOUT_SiX  Ú%DEFAULT_WORKER_HEALTH_CHECK_TIMEOUT_SÚ&RAY_TRAIN_WORKER_GROUP_START_TIMEOUT_Sg      N@Ú$DEFAULT_WORKER_GROUP_START_TIMEOUT_SÚRAY_TRAIN_COLLECTIVE_TIMEOUT_SéÿÿÿÿÚDEFAULT_COLLECTIVE_TIMEOUT_SÚ$RAY_TRAIN_COLLECTIVE_WARN_INTERVAL_Sé<   Ú"DEFAULT_COLLECTIVE_WARN_INTERVAL_SÚRAY_TRAIN_ENABLE_PRINT_PATCHTÚRAY_TRAIN_V2_ENABLEDÚ.RAY_TRAIN_ENABLE_CONTROLLER_STRUCTURED_LOGGINGÚ*RAY_TRAIN_ENABLE_WORKER_STRUCTURED_LOGGINGÚ+RAY_TRAIN_ENABLE_STATE_ACTOR_RECONCILIATIONÚ/RAY_TRAIN_STATE_ACTOR_RECONCILIATION_INTERVAL_Sg      >@Ú-DEFAULT_STATE_ACTOR_RECONCILIATION_INTERVAL_Sé   ÚGET_ACTOR_TIMEOUT_Sé   Ú!CONTROLLERS_TO_POLL_PER_ITERATIONÚRAY_TRAIN_CALLBACKSÚ0ÚRAY_TRAIN_METRICS_ENABLEDÚreturnc                   C   s
   t tdƒS )NT)r   ÚV2_ENABLED_ENV_VAR© r$   r$   úV/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/train/v2/_internal/constants.pyÚis_v2_enabledx   s   
r&   c                  C   s(   i } t D ]}t|ƒrtj| | |< q| S )aN  Returns a dictionary of environment variables that should be propagated
    from the driver to the controller, and then from the controller
    to each training worker.

    This way, users only need to set environment variables in one place
    when launching the script instead of needing to manually set a runtime environment.
    )ÚENV_VARS_TO_PROPAGATEr   ÚosÚenviron)Úenv_varsÚenv_varr$   r$   r%   Úget_env_vars_to_propagate|   s   €r,   )/r(   Útypingr   Úray._common.constantsr   Úray._private.ray_constantsr   r   Ú_UNSUPPORTEDÚ_DEPRECATEDÚ VALIDATE_STORAGE_MARKER_FILENAMEÚ$CHECKPOINT_MANAGER_SNAPSHOT_FILENAMEÚAWS_RETRYABLE_TOKENSÚHEALTH_CHECK_INTERVAL_S_ENV_VARr	   ÚfloatÚ__annotations__Ú%WORKER_HEALTH_CHECK_TIMEOUT_S_ENV_VARr   Ú$WORKER_GROUP_START_TIMEOUT_S_ENV_VARr   ÚCOLLECTIVE_TIMEOUT_S_ENV_VARr   Ú"COLLECTIVE_WARN_INTERVAL_S_ENV_VARr   ÚENABLE_PRINT_PATCH_ENV_VARÚDEFAULT_ENABLE_PRINT_PATCHr#   Ú,ENABLE_CONTROLLER_STRUCTURED_LOGGING_ENV_VARÚ(ENABLE_WORKER_STRUCTURED_LOGGING_ENV_VARÚ!DEFAULT_ENABLE_CONTROLLER_LOGGINGÚDEFAULT_ENABLE_WORKER_LOGGINGÚ)ENABLE_STATE_ACTOR_RECONCILIATION_ENV_VARÚ)DEFAULT_ENABLE_STATE_ACTOR_RECONCILIATIONÚ-STATE_ACTOR_RECONCILIATION_INTERVAL_S_ENV_VARr   r   Úintr   ÚRAY_TRAIN_CALLBACKS_ENV_VARÚ0DEFAULT_RAY_WARN_BLOCKING_GET_INSIDE_ASYNC_VALUEr'   ÚMETRICS_ENABLED_ENV_VARÚboolr&   Ústrr,   r$   r$   r$   r%   Ú<module>   sj   
 ÿÿÿô