&`i_ddlZddlZddlmZmZmZddlZddlm Z ddl m Z ddl m Z ejeZdZ de defd ZGd d Ze Gd d e ZdS)N)DictListOptional)Searcher)+_set_search_properties_backwards_compatible) PublicAPI__trial_index__searcher num_samplesct|tr&||jzrtddSdSdS)Na`num_samples` is now expected to be the total number of trials, including the repeat trials. For example, set num_samples=15 if you intend to obtain 3 search algorithm suggestions and repeat each suggestion 5 times. Any leftover trials (num_samples mod repeat) will be ignored.) isinstanceRepeaterrepeatloggerwarning)r r s l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/search/repeater.py_warn_num_samplesrs[(H%% +*G  8         ceZdZdZddededefdZdefdZd e fd Z ded e fd Z d e fd Z d eee fdZd efdZdS) _TrialGroupaInternal class for grouping trials of same parameters. This is used when repeating trials for reducing training variance. Args: primary_trial_id: Trial ID of the "primary trial". This trial is the one that the Searcher is aware of. config: Suggested configuration shared across all trials in the trial group. max_trials: Max number of trials to execute within this group. primary_trial_idconfig max_trialsct|tusJd|||_||_|di|_||_dS)Nzconfig is not a dict, got {})typedictformatrr_trialsr)selfrrrs r__init__z_TrialGroup.__init__*sVF||t###%C%J%J6%R%R### 0 ($/ $rtrial_idczt|j|jksJ|j|ddSN)lenrr setdefault)r r"s raddz_TrialGroup.add1s=4<  4?2222 $/////rreturnc<t|j|jkSr$)r%rrr s rfullz_TrialGroup.full5s4<  DO33rscorecR||jvsJ|td||j|<dS)Nz%Internal Error: Score cannot be None.)r ValueError)r r"r,s rreportz_TrialGroup.report8s;4<'''' =DEE E!& Xrcrd|jvot|j|jkSr$)rvaluesr%rr*s rfinished_reportingz_TrialGroup.finished_reporting>s3  ++-- - V#dl2C2Ct2V rcNt|jSr$)listrr1r*s rscoresz_TrialGroup.scoresCsDL''))***rc*t|jSr$)r%rr*s rcountz_TrialGroup.countFs4<   rN)r)__name__ __module__ __qualname____doc__strrintr!r'boolr+floatr/r2rrr5r7rrrrs  %%%d%%%%%0C00004d4444's'5''''  D    +Xe_-++++!s!!!!!!rrceZdZdZddededeffd Zded e e fd Z dded e e fd Z d e fdZ de fdZdefdZdefdZde ede ede d efdZxZS)raA wrapper algorithm for repeating trials of same parameters. Set tune.TuneConfig(num_samples=...) to be a multiple of `repeat`. For example, set num_samples=15 if you intend to obtain 3 search algorithm suggestions and repeat each suggestion 5 times. Any leftover trials (num_samples mod repeat) will be ignored. It is recommended that you do not run an early-stopping TrialScheduler simultaneously. Args: searcher: Searcher object that the Repeater will optimize. Note that the Searcher will only see 1 trial among multiple repeated trials. The result/metric passed to the Searcher upon trial completion will be averaged among all repeats. repeat: Number of times to generate a trial with a repeated configuration. Defaults to 1. set_index: Sets a tune.search.repeater.TRIAL_INDEX in Trainable/Function config which corresponds to the index of the repeated trial. This can be used for seeds. Defaults to True. Example: .. code-block:: python from ray.tune.search import Repeater search_alg = BayesOptSearch(...) re_search_alg = Repeater(search_alg, repeat=10) # Repeat 2 samples 10 times each. tuner = tune.Tuner( trainable, tune_config=tune.TuneConfig( search_alg=re_search_alg, num_samples=20, ), ) tuner.fit() rTr r set_indexc||_||_||_g|_i|_d|_t t||jj |jj dS)N)metricmode) r r _set_index_groups_trial_id_to_group_current_groupsuperrr!rDrE)r r rrB __class__s rr!zRepeater.__init__wso   # "$" h&&='dm.@ '     rr"r(c|j|jrn|j|}||St |t j||j|_|j |jd}n3|j }|j ||jj }|j r ||t<|j|j|<|S)N)rr)rIr+r suggestrcopydeepcopyrrGappendr7r'rrF TRIAL_INDEXrH)r r"rindex_in_groups rrMzRepeater.suggests   &$*=*B*B*D*D &]**844F~ "-$-//DK###D  L   3 4 4 4NN!06688N   # #H - - -$+0022 ? 1"0F; ,0,?) rNresultc ||jvr@td|t |j|j|}|r|jj|vr tj}n||jj}| ||| rN| }|jj |j fd|jjtj|ii|dSdS)aStores the score for and keeps track of a completed trial. Stores the metric of a trial as nan if any of the following conditions are met: 1. ``result`` is empty or not provided. 2. ``result`` is provided but no metric was provided. z;Trial {} not in group; cannot report score. Seen trials: {}rSN)rHrerrorrr4r rDnpnanr/r2r5on_trial_completernanmean)r r"rSkwargs trial_groupr,r5s rrXzRepeater.on_trial_completes 42 2 2 LL""(&48O3P3P"Q"Q   -h7  1-V;;FEE4=/0E8U+++  ) ) + +  ''))F +DM +,   ,bj.@.@A        rc>|j}|d=|S)Nr )__dict__rN)r self_states r get_statezRepeater.get_states#]'')) z "rstatec:|j|dSr$)r]update)r r`s r set_statezRepeater.set_states U#####rcheckpoint_pathc:|j|dSr$)r saver rds rrfz Repeater.saves ?+++++rc:|j|dSr$)r restorergs rrizRepeater.restores o.....rrDrErc 6t|jj|||fi|Sr$)rr set_search_properties)r rDrErspecs rrkzRepeater.set_search_propertiess1; M /v  IM   r)rTr$)r8r9r:r;rr=r>r!r<rrrMrXr_rcrfrirk __classcell__)rKs@rrrJs[))V     3  t       (#x~<4 $t$$$$,C,,,,/s//// sm +3C= BF         rr)rNloggingtypingrrrnumpyrVray.tune.search.searcherrray.tune.search.utilrray.utilr getLoggerr8rrQr=rrrr@rrrus& ''''''''''------LLLLLL  8 $ $ G  s    +!+!+!+!+!+!+!+!\ | | | | | x| |  | | | r