:jddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z erddl m Z e d Gd d Z ed d dGdde Zy)) annotations)Any) TYPE_CHECKINGN)deprecated_class)experimental_class)experimental_func) FrozenTrial2.8.0c~eZdZdZ d d dZd dZeedd dZeedd dZ y) RetryHeartbeatStaleTrialCallbackaRetry a heartbeat-stale trial up to a maximum number of times. When a running trial becomes stale due to the heartbeat mechanism, this callback can be used with a class in :mod:`optuna.storages` to recreate the trial in ``TrialState.WAITING`` to queue up the trial to be run again. The original stale trial can be identified by the :func:`~optuna.storages.RetryHeartbeatStaleTrialCallback.retried_trial_number` function. Even if repetitive failure occurs (a retried trial becomes stale again), this method returns the number of the original trial. To get a full list including the numbers of the retried trials as well as their original trial, call the :func:`~optuna.storages.RetryHeartbeatStaleTrialCallback.retry_history` function. This callback is helpful in environments where running trials may become stale due to external conditions, such as worker preemption or unexpected process termination. Usage: .. testcode:: import optuna from optuna.storages import RetryHeartbeatStaleTrialCallback storage = optuna.storages.RDBStorage( url="sqlite:///:memory:", heartbeat_interval=60, grace_period=120, heartbeat_stale_trial_callback=RetryHeartbeatStaleTrialCallback(max_retry=3), ) study = optuna.create_study( storage=storage, ) .. seealso:: See :class:`~optuna.storages.RDBStorage`. Args: max_retry: The max number of times a trial can be retried. Must be set to :obj:`None` or an integer. If set to the default value of :obj:`None` will retry indefinitely. If set to an integer, will only retry that many times. inherit_intermediate_values: Option to inherit `trial.intermediate_values` reported by :func:`optuna.trial.Trial.report` from the failed trial. Default is :obj:`False`. Nc ||_||_y)N) _max_retry_inherit_intermediate_values)self max_retryinherit_intermediate_valuess o/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/optuna/storages/_callbacks.py__init__z)RetryHeartbeatStaleTrialCallback.__init__As$,G)c |jgd|j}|dj|j|j|jt |dkry|j t jt jjj|j|j|j||jr |jndy)N) failed_trial retry_historyr)stateparams distributions user_attrs system_attrsintermediate_values)numberrappendrlen add_trialoptuna create_trialtrial TrialStateWAITINGrrrrr)rstudyr%rs r__call__z)RetryHeartbeatStaleTrialCallback.__call__Gs!LL(   ( _%,,U\\: ?? &\/%B!CC    ll--55||#11 ++)151R1RE--X\  rr c:|jjddS)a Return the number of the original trial being retried. Args: trial: The trial object. Returns: The number of the first failed trial. If not retry of a previous trial, returns :obj:`None`. rNrgetr%s rretried_trial_numberz5RetryHeartbeatStaleTrialCallback.retried_trial_number_s!!%%nd;;rz3.0.0c:|jjdgS)atReturn the list of retried trial numbers with respect to the specified trial. Args: trial: The trial object. Returns: A list of trial numbers in ascending order of the series of retried trials. The first item of the list indicates the original trial which is identical to the :func:`~optuna.storages.RetryHeartbeatStaleTrialCallback.retried_trial_number`, and the last item is the one right before the specified trial in the retry series. If the specified trial is not a retry of any trial, returns an empty list. rr+r-s rrz.RetryHeartbeatStaleTrialCallback.retry_historyos !!%%or::r)NF)r int | NonerboolreturnNone)r(z'optuna.study.Study'r%r r2r3)r%r r2r0)r%r r2z list[int]) __name__ __module__ __qualname____doc__rr) staticmethodrr.rrrr r sw-`QVH#HIMH H  0w <  <w; ;rr z4.9.0z6.0.0z/Use `RetryHeartbeatStaleTrialCallback` instead.)textceZdZdZy)RetryFailedTrialCallbackzDeprecated alias of :class:`~optuna.storages.RetryHeartbeatStaleTrialCallback`. Deprecated in v4.9.0. This class will be removed in v6.0.0. N)r4r5r6r7r9rrr<r<s   rr<) __future__rtypingrrr#optuna._deprecatedroptuna._experimentalrr optuna.trialr r r<r9rrrBsn" /32(Gn;n;n;b  :  ?   r