o
    c²“iÞ  ã                	   @   sð   U d dl 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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ee
eeeeeeeh	ZdZdZ dZ!de"fd d!„Z#dee$e$f fd"d#„Z%dS )$é    N)ÚDict)Úenv_boolÚenv_set_by_userÚUNSUPPORTEDÚ
DEPRECATEDz.validate_storage_markerz checkpoint_manager_snapshot.jsonÚ!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      >@Ú$DEFAULT_WORKER_GROUP_START_TIMEOUT_SÚ"RAY_TRAIN_REPORT_BARRIER_TIMEOUT_Si  Ú DEFAULT_REPORT_BARRIER_TIMEOUT_SÚ(RAY_TRAIN_REPORT_BARRIER_WARN_INTERVAL_Sé<   Ú&DEFAULT_REPORT_BARRIER_WARN_INTERVAL_SÚRAY_TRAIN_ENABLE_PRINT_PATCHÚ1ÚRAY_TRAIN_V2_ENABLEDÚ.RAY_TRAIN_ENABLE_CONTROLLER_STRUCTURED_LOGGINGÚ*RAY_TRAIN_ENABLE_WORKER_STRUCTURED_LOGGINGÚRAY_TRAIN_CALLBACKSÚRAY_TRAIN_METRICS_ENABLEDÚ!RAY_TRAIN_RUN_CONTROLLER_AS_ACTORÚreturnc                   C   s
   t tdƒS )NF)r   ÚV2_ENABLED_ENV_VAR© r   r   úT/home/ubuntu/.local/lib/python3.10/site-packages/ray/train/v2/_internal/constants.pyÚis_v2_enabled\   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._private.ray_constantsr   r   Ú_UNSUPPORTEDÚ_DEPRECATEDÚ VALIDATE_STORAGE_MARKER_FILENAMEÚ$CHECKPOINT_MANAGER_SNAPSHOT_FILENAMEÚHEALTH_CHECK_INTERVAL_S_ENV_VARr   ÚfloatÚ__annotations__Ú%WORKER_HEALTH_CHECK_TIMEOUT_S_ENV_VARr
   Ú$WORKER_GROUP_START_TIMEOUT_S_ENV_VARr   Ú REPORT_BARRIER_TIMEOUT_S_ENV_VARr   Ú&REPORT_BARRIER_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ÚRAY_TRAIN_CALLBACKS_ENV_VARr   ÚMETRICS_ENABLED_ENV_VARÚRUN_CONTROLLER_AS_ACTOR_ENV_VARÚDEFAULT_RUN_CONTROLLER_AS_ACTORÚboolr   Ústrr$   r   r   r   r   Ú<module>   sR   
 ÿ÷