o $is @sddlmZmZddlmZddlmZmZddl m Z m Z ddl m Z mZddlmZe eGdddeZed d ed d d edeedd fddZed d edeefddZdefddZd S))DictOptional) Checkpoint)_warn_session_misuse get_session)V2_MIGRATION_GUIDE_MESSAGE_v2_migration_warnings_enabled) _copy_doc_log_deprecation_warning) PublicAPIc@s eZdZdS)rN)__name__ __module__ __qualname__rrb/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/tune/trainable/trainable_fn_utils.pyr srstable) stabilityN checkpointmetricsrreturncCs6|rt|tstrtdttj||ddS)a`Report metrics and optionally save and register a checkpoint to Ray Tune. If a checkpoint is provided, it will be :ref:`persisted to storage `. .. note:: Each invocation of this method will automatically increment the underlying ``training_iteration`` number. The physical meaning of this "iteration" is defined by user depending on how often they call ``report``. It does not necessarily map to one epoch. Args: metrics: The metrics you want to report. checkpoint: The optional checkpoint you want to report. zThe `Checkpoint` class should be imported from `ray.tune` when passing it to `ray.tune.report` in a Tune function. Please update your imports. rN) isinstancerrr rrreport)rrrrrrsrcCstjS)zCAccess the latest reported checkpoint to resume from if one exists.)rloaded_checkpointrrrrget_checkpoint3srcCstotjduS)N)r world_rankrrrr_in_tune_session;sr)typingrrray.train._checkpointrTrainCheckpointray.train._internal.sessionrrray.train.constantsrrray.train.utilsr r ray.util.annotationsr rrboolrrrrrs  $