
    &`iu                     n   U d dl mZ 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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eed<   dZde dZdefdZ dZ!dZ"dZ#dZ$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*e+e,e.hZ0d%Z1d&Z2dS )'    )Path)AnyN)env_bool)0COPY_DIRECTORY_CHECKPOINTS_INSTEAD_OF_MOVING_ENVEVALUATION_DATASET_KEY	MODEL_KEYPREPROCESSOR_KEYTRAIN_DATASET_KEYreturnc                      t          j                    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     g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/train/constants.py_get_ray_train_session_dirr      sT    RR!RRR(==?? hjjr   z~/ray_resultscheckpoint_dir_name_time_total_s	_hostname_node_ip_pidzresults.jsoncheckpoints_current_checkpoint_id
DEPRECATED_DEPRECATED_VALUE&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                  ,    t          t          d          S )NT)r   $ENABLE_V2_MIGRATION_WARNINGS_ENV_VARr   r   r   _v2_migration_warnings_enabledr%   D   s    8$???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*TUNE_ONLY_STORE_CHECKPOINT_SCORE_ATTRIBUTE&TORCH_PROCESS_GROUP_SHUTDOWN_TIMEOUT_S   "JAX_DISTRIBUTED_SHUTDOWN_TIMEOUT_Scheckpoint_metadatacheckpoint_rank)3pathlibr   typingr   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_IDr!   __annotations__r$   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/   r0   r1   .DEFAULT_TORCH_PROCESS_GROUP_SHUTDOWN_TIMEOUT_Sr3   *DEFAULT_JAX_DISTRIBUTED_SHUTDOWN_TIMEOUT_STRAIN_ENV_VARSCHECKPOINT_METADATA_KEYCHECKPOINT_RANK_KEYr   r   r   <module>rU      s3                



 / / / / / /             C     tO,,7799BBDD  , 
 ,^ZV  "  ( 
 . 
 & 3 % % % (P $0 ,0 0 0 @ @ @ @ @ 6 ' )R % (P $
 2 * +Y ' 'H # ">  2  )P % #D  1 + *R &13 . &J "-/ * +)-'")#.*& 0  (   r   