o
    1 iu                     @   s   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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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                   C   s(   t  sJ dtt jjj d S )Nz/Ray must be initialized to get the session dir.Z	artifacts)rayZis_initializedr   Z_privateZworkerZ_global_nodeZget_session_dir_pathas_posix r   r   _/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/train/constants.py_get_ray_train_session_dir   s   r   z~/ray_resultsZcheckpoint_dir_nameZ_time_total_sZ	_hostnameZ_node_ip_pidzresults.jsonZcheckpointsZ_current_checkpoint_idZ
DEPRECATED_DEPRECATED_VALUEZ&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_enabledD   s   
r   Z/TRAIN_RESULT_ENABLE_DETAILED_AUTOFILLED_METRICSZ'TRAIN_ENABLE_SHARE_CUDA_VISIBLE_DEVICESZ&TRAIN_ENABLE_SHARE_HIP_VISIBLE_DEVICESZ+TRAIN_ENABLE_SHARE_NEURON_CORES_ACCELERATORZ,TRAIN_ENABLE_SHARE_ASCEND_RT_VISIBLE_DEVICESZTRAIN_PLACEMENT_GROUP_TIMEOUT_SZ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_SZcheckpoint_metadataZcheckpoint_rank)3pathlibr   typingr   r   Zray._private.ray_constantsr   Zray.air.constantsr   r   r   r   r	   strr   
expanduserr   ZDEFAULT_STORAGE_PATHZCHECKPOINT_DIR_NAMEZTIME_TOTAL_SZWORKER_HOSTNAMEZWORKER_NODE_IPZ
WORKER_PIDZDETAILED_AUTOFILLED_KEYSZRESULT_FILE_JSONZTRAIN_CHECKPOINT_SUBDIRZTUNE_CHECKPOINT_IDr   __annotations__r   ZV2_MIGRATION_GUIDE_MESSAGEboolr   Z&ENABLE_DETAILED_AUTOFILLED_METRICS_ENVZ%ENABLE_SHARE_CUDA_VISIBLE_DEVICES_ENVZ$ENABLE_SHARE_HIP_VISIBLE_DEVICES_ENVZ)ENABLE_SHARE_NEURON_CORES_ACCELERATOR_ENVZ'ENABLE_SHARE_NPU_RT_VISIBLE_DEVICES_ENVZ#TRAIN_PLACEMENT_GROUP_TIMEOUT_S_ENVZTRAIN_ENABLE_WORKER_SPREAD_ENVr   r   r   r   r   Z.DEFAULT_TORCH_PROCESS_GROUP_SHUTDOWN_TIMEOUT_Sr   Z*DEFAULT_JAX_DISTRIBUTED_SHUTDOWN_TIMEOUT_SZTRAIN_ENV_VARSZCHECKPOINT_METADATA_KEYZCHECKPOINT_RANK_KEYr   r   r   r   <module>   sn    			