jUddlmZddlZddlmZddlmZddlmZddlmZ ddl m Z erddl m Z dad ed <Gd d ejZGd dZdS)) annotationsN)Any) TYPE_CHECKING)tqdm)logging) optuna_warn)Studyz_TqdmLoggingHandler | None _tqdm_handlerceZdZddZdS)_TqdmLoggingHandlerrecordrreturnNonec ||}tj||dS#tt f$rt $r||YdSwxYw)N)formatrwriteflushKeyboardInterrupt SystemExit Exception handleError)selfr msgs v/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/optuna/progress_bar.pyemitz_TqdmLoggingHandler.emits %++f%%C JsOOO JJLLLLL!:.     % % %   V $ $ $ $ $ $ %s=A0A54A5N)r rrr)__name__ __module__ __qualname__rrr r s(%%%%%%r r c0eZdZdZ ddd ZddZddZdS) _ProgressBaraPProgress Bar implementation for :func:`~optuna.study.Study.optimize` on the top of `tqdm`. Args: is_valid: Whether to show progress bars in :func:`~optuna.study.Study.optimize`. n_trials: The number of trials. timeout: Stop study after the given number of second(s). Nis_validbooln_trials int | Nonetimeout float | Nonerrc|r||td|o|p|du|_||_||_d|_|jr|jt |j|_nC|j:t j|j}d|z}t |j||_nJta t tj t tjtjtjtdSdS)NzFProgress bar won't be displayed because n_trials and timeout are None.g)totalz+{desc} {percentage:3.0f}%|{bar}| {elapsed}/)r* bar_format)r _is_valid _n_trials_timeout_last_elapsed_secondsr _progress_barformat_intervalr r setLevelrINFO setFormatteroptuna_loggingcreate_default_formatterdisable_default_handler_get_library_root_logger addHandler)rr#r%r'r*fmts r__init__z_ProgressBar.__init__,s=  b(W_ ` a a a!Gx':74&G! %(" > P~)%)%?%?%?""*,T];;CeK%) #%N%N%N"" 011M  " "7< 0 0 0  & &~'N'P'P Q Q Q  2 4 4 4  3 5 5 @ @ O O O O O! P Pr elapsed_secondsfloatstudyr c|jr|sD d|jjd|jd}|j|n#t$rYnwxYw|jK|j d|j (|j |dd|j ddSdS|j E||j z }||j kr |||j z z}|j |||_ dSJdS) zUpdate the progress bars if ``is_valid`` is :obj:`True`. Args: elapsed_seconds: The time past since :func:`~optuna.study.Study.optimize` started. study: The current study object. z Best trial: z. Best value: z.6gNz.02f/z seconds) r,_is_multi_objective best_trialnumber best_valuer0set_description ValueErrorr-updater.set_postfix_strr/)rr<r>r time_diffs rrHz_ProgressBar.updateLsl > ,,.. >u'7'>>>','7=>> &66s;;;;!D~)"))!,,,=,&66*III$-III-, *+d.HH "T]224=!@@I")))444-<*** =  s2A AAc|jrb|jtJt jtt jdSdS)zClose progress bars.N)r,r0closer r5r8 removeHandlerenable_default_handler)rs rrLz_ProgressBar.closevsk > 4   $ $ & & & ,,,  3 5 5 C CM R R R  1 3 3 3 3 3  4 4r )NN)r#r$r%r&r'r(rr)r<r=r>r rr)rr)rrr__doc__r;rHrLrr rr"r" sn   $ $ PPPPP@((((T444444r r") __future__rrtypingrr tqdm.autoroptunar5optuna._warningsr optuna.studyr r __annotations__ StreamHandlerr r"rr rrXs""""""" ,,,,,,((((((#"""""",0 0000 % % % % %'/ % % %]4]4]4]4]4]4]4]4]4]4r