o $i/@sUddlZddlmZmZmZddlmZddlmZm Z ddl m Z ddl mZddlmZddlmZmZdaed ed <eZd eZed d Gdd d e Zed d defddZdS)N)AnyDictOptional)session)V2_MIGRATION_GUIDE_MESSAGE_v2_migration_warnings_enabled) TrainContext) _copy_doc)PlacementGroupFactory) Deprecated PublicAPI TuneContext _tune_contextz`{}` is deprecated for Ray Tune because there is no concept of worker ranks for Ray Tune, so these methods only make sense to use in the context of a Ray Train worker. beta) stabilityc@sfeZdZdZeejdefddZeejdefddZeej de fddZ eej defd d Z e de eeffd d Ze ed edeejdefddZe ededeejdefddZe ededeejdefddZe ededeejdefddZe ededeejdefddZdS)r z5Context to access metadata within Ray Tune functions.returncCtSN)rget_trial_nameselfrM/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/tune/context.pyr"zTuneContext.get_trial_namecCrr)r get_trial_idrrrrr&rzTuneContext.get_trial_idcCrr)rget_trial_resourcesrrrrr*rzTuneContext.get_trial_resourcescCrr)r get_trial_dirrrrrr.rzTuneContext.get_trial_dircCstd)NzG`get_metadata` is deprecated for Ray Tune, as it has never been usable.)DeprecationWarningrrrr get_metadata4szTuneContext.get_metadataget_world_size)messagewarningcCrr)rrrrrrr:zTuneContext.get_world_sizeget_world_rankcCrr)rr#rrrrr#Br"zTuneContext.get_world_rankget_local_rankcCrr)rr$rrrrr$Jr"zTuneContext.get_local_rankget_local_world_sizecCrr)rr%rrrrr%Rsz TuneContext.get_local_world_size get_node_rankcCrr)rr&rrrrr&\r"zTuneContext.get_node_rankN)__name__ __module__ __qualname____doc__r rrstrrrr rr rrr+_TRAIN_SPECIFIC_CONTEXT_DEPRECATION_MESSAGEformatrTrainV1Contextrintr#r$r%r&rrrrr sXrcCs:ttdur tatWdS1swYdS)zGet or create a singleton Ray Tune context. The context is only available in a tune function passed to the `ray.tune.Tuner`. See the :class:`~ray.tune.TuneContext` API reference to see available methods. N)_tune_context_lockrr rrrr get_contextes  $r1) threadingtypingrrrray.train._internalrray.train.constantsrrray.train.contextrr.ray.train.utilsr #ray.tune.execution.placement_groupsr ray.util.annotationsr r r__annotations__Lockr0r,r r1rrrrs$     J