o
    ci                     @   s   d dl mZ d dlZd dlmZ d dlmZmZmZm	Z	m
Z
 defddZed  Zd	Zd
ZdZdZdZeeeehZdZdZdZdZdZde dZdef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e!e#e$e%e&e'hZ(d!Z)d"Z*dS )#    )PathN)env_bool)0COPY_DIRECTORY_CHECKPOINTS_INSTEAD_OF_MOVING_ENVEVALUATION_DATASET_KEY	MODEL_KEYPREPROCESSOR_KEYTRAIN_DATASET_KEYreturnc                   C   s(   t  sJ dtt jjj d S )Nz/Ray must be initialized to get the session dir.	artifacts)rayis_initializedr   _privateworker_global_nodeget_session_dir_pathas_posix r   r   G/home/ubuntu/.local/lib/python3.10/site-packages/ray/train/constants.py_get_ray_train_session_dir   s   r   z~/ray_resultscheckpoint_dir_name_time_total_s	_hostname_node_ip_pidzresults.jsoncheckpoints_current_checkpoint_id
DEPRECATED&RAY_TRAIN_ENABLE_V2_MIGRATION_WARNINGSzSee this issue for more context and migration options: https://github.com/ray-project/ray/issues/49454. Disable these warnings by setting the environment variable: z=0c                   C   s
   t tdS )NT)r   $ENABLE_V2_MIGRATION_WARNINGS_ENV_VARr   r   r   r   _v2_migration_warnings_enabledA   s   
r   /TRAIN_RESULT_ENABLE_DETAILED_AUTOFILLED_METRICS'TRAIN_ENABLE_SHARE_CUDA_VISIBLE_DEVICES&TRAIN_ENABLE_SHARE_HIP_VISIBLE_DEVICES+TRAIN_ENABLE_SHARE_NEURON_CORES_ACCELERATOR,TRAIN_ENABLE_SHARE_ASCEND_RT_VISIBLE_DEVICESTRAIN_PLACEMENT_GROUP_TIMEOUT_STRAIN_ENABLE_WORKER_SPREADRAY_CHDIR_TO_TRIAL_DIR%RAY_TRAIN_COUNT_PREEMPTION_AS_FAILURERAY_TRAIN_ENABLE_STATE_TRACKINGcheckpoint_metadatacheckpoint_rank)+pathlibr   r   ray._private.ray_constantsr   ray.air.constantsr   r   r   r   r   strr   
expanduserr   DEFAULT_STORAGE_PATHCHECKPOINT_DIR_NAMETIME_TOTAL_SWORKER_HOSTNAMEWORKER_NODE_IP
WORKER_PIDDETAILED_AUTOFILLED_KEYSRESULT_FILE_JSONTRAIN_CHECKPOINT_SUBDIRTUNE_CHECKPOINT_ID_DEPRECATED_VALUEr   V2_MIGRATION_GUIDE_MESSAGEboolr   &ENABLE_DETAILED_AUTOFILLED_METRICS_ENV%ENABLE_SHARE_CUDA_VISIBLE_DEVICES_ENV$ENABLE_SHARE_HIP_VISIBLE_DEVICES_ENV)ENABLE_SHARE_NEURON_CORES_ACCELERATOR_ENV'ENABLE_SHARE_NPU_RT_VISIBLE_DEVICES_ENV#TRAIN_PLACEMENT_GROUP_TIMEOUT_S_ENVTRAIN_ENABLE_WORKER_SPREAD_ENVr'   r(   r)   TRAIN_ENV_VARSCHECKPOINT_METADATA_KEYCHECKPOINT_RANK_KEYr   r   r   r   <module>   sZ    		