jn'&UddlmZddlZddlmZddlmZddlmZddlmZddlmZddlmZdd lm Z ddl Z ddl Z ddl Z gd Z e jZd ed <dad ed<d#dZd$dZd%dZd&dZd&dZd'dZd(dZd)dZd&dZd&d Zd&d!Zd&d"ZdS)*) annotationsN)CRITICAL)DEBUG)ERROR)FATAL)INFO)WARN)WARNING)rrrrrr r zthreading.Lock_lockzlogging.Handler | None_default_handlerreturnlogging.FormattercVd}d}tjd|d|tjS)z}Create a default formatter of log messages. This function is not supposed to be directly accessed by library users. z[%(levelname)1.1s %(asctime)s]z %(message)sz %(log_color)sz %(reset)s )stream)colorlogTTYColoredFormattersysstderr)headermessages q/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/optuna/logging.pycreate_default_formatterrs@ .FG  '333'33z   strcBtddS)N.r)__name__splitrr_get_library_namer ,s >>#  q !!rlogging.LoggercBtjtSN)logging getLoggerr rrr_get_library_root_loggerr&0s  .00 1 11rNonec~t5tr ddddStjatt t }|t|tj d|_ ddddS#1swxYwYdS)NF) r r r$ StreamHandler setFormatterrr& addHandlersetLevelr propagatelibrary_root_loggers r_configure_library_root_loggerr04s  . .    . . . . . . . .#022%%&>&@&@AAA/G.H.H&&'7888$$W\222(-% . . . . . . . . . . . . . . . . . .s B2BB22B69B6ct5ts ddddSt}|t|t jdaddddS#1swxYwYdSr#)r r r& removeHandlerr,r$NOTSETr.s r_reset_library_root_loggerr4Es             /G.H.H))*:;;;$$W^444                  s A4A A44A8;A8namecFttj|S)zzReturn a logger with the specified name. This function is not supposed to be directly accessed by library users. )r0r$r%)r5s r get_loggerr7Rs! #$$$  T " ""rintc^ttS)aReturn the current level for the Optuna's root logger. Example: Get the default verbosity level. .. testsetup:: def objective(trial): x = trial.suggest_float("x", -100, 100) y = trial.suggest_categorical("y", [-1, 0, 1]) return x**2 + y .. testcode:: import optuna # The default verbosity level of Optuna is `optuna.logging.INFO`. print(optuna.logging.get_verbosity()) # 20 print(optuna.logging.INFO) # 20 # There are logs of the INFO level. study = optuna.create_study() study.optimize(objective, n_trials=5) # [I 2021-10-31 05:35:17,232] A new study created ... # [I 2021-10-31 05:35:17,238] Trial 0 finished with value: ... # [I 2021-10-31 05:35:17,245] Trial 1 finished with value: ... # ... .. testoutput:: :hide: 20 20 Returns: Logging level, e.g., ``optuna.logging.DEBUG`` and ``optuna.logging.INFO``. .. note:: Optuna has following logging levels: - ``optuna.logging.CRITICAL``, ``optuna.logging.FATAL`` - ``optuna.logging.ERROR`` - ``optuna.logging.WARNING``, ``optuna.logging.WARN`` - ``optuna.logging.INFO`` - ``optuna.logging.DEBUG`` )r0r&getEffectiveLevelrrr get_verbosityr;\s)d#$$$ # % % 7 7 9 99r verbositycdtt|dS)aaSet the level for the Optuna's root logger. Example: Set the logging level ``optuna.logging.WARNING``. .. testsetup:: def objective(trial): x = trial.suggest_int("x", -10, 10) return x**2 .. testcode:: import optuna # There are INFO level logs. study = optuna.create_study() study.optimize(objective, n_trials=10) # [I 2021-10-31 02:59:35,088] Trial 0 finished with value: 16.0 ... # [I 2021-10-31 02:59:35,091] Trial 1 finished with value: 1.0 ... # [I 2021-10-31 02:59:35,096] Trial 2 finished with value: 1.0 ... # Setting the logging level WARNING, the INFO logs are suppressed. optuna.logging.set_verbosity(optuna.logging.WARNING) study.optimize(objective, n_trials=10) .. testcleanup:: optuna.logging.set_verbosity(optuna.logging.INFO) Args: verbosity: Logging level, e.g., ``optuna.logging.DEBUG`` and ``optuna.logging.INFO``. .. note:: Optuna has following logging levels: - ``optuna.logging.CRITICAL``, ``optuna.logging.FATAL`` - ``optuna.logging.ERROR`` - ``optuna.logging.WARNING``, ``optuna.logging.WARN`` - ``optuna.logging.INFO`` - ``optuna.logging.DEBUG`` N)r0r&r,)r<s r set_verbosityr>s1^#$$$'' 22222rcttJttdS)aDisable the default handler of the Optuna's root logger. Example: Stop and then resume logging to :obj:`sys.stderr`. .. testsetup:: def objective(trial): x = trial.suggest_float("x", -100, 100) y = trial.suggest_categorical("y", [-1, 0, 1]) return x**2 + y .. testcode:: import optuna study = optuna.create_study() # There are no logs in sys.stderr. optuna.logging.disable_default_handler() study.optimize(objective, n_trials=10) # There are logs in sys.stderr. optuna.logging.enable_default_handler() study.optimize(objective, n_trials=10) # [I 2020-02-23 17:00:54,314] Trial 10 finished with value: ... # [I 2020-02-23 17:00:54,356] Trial 11 finished with value: ... # ... N)r0r r&r2rrrdisable_default_handlerr@s>B#$$$  ' ' ',,-=>>>>>rcttJttdS)zEnable the default handler of the Optuna's root logger. Please refer to the example shown in :func:`~optuna.logging.disable_default_handler()`. N)r0r r&r+rrrenable_default_handlerrBs= #$$$  ' ' '))*:;;;;;rcHtdt_dS)aDisable propagation of the library log outputs. Note that log propagation is disabled by default. You only need to use this function to stop log propagation when you use :func:`~optuna.logging.enable_propagation()`. Example: Stop propagating logs to the root logger on the second optimize call. .. testsetup:: def objective(trial): x = trial.suggest_float("x", -100, 100) y = trial.suggest_categorical("y", [-1, 0, 1]) return x**2 + y .. testcode:: import optuna import logging optuna.logging.disable_default_handler() # Disable the default handler. logger = logging.getLogger() logger.setLevel(logging.INFO) # Setup the root logger. logger.addHandler(logging.FileHandler("foo.log", mode="w")) optuna.logging.enable_propagation() # Propagate logs to the root logger. study = optuna.create_study() logger.info("Logs from first optimize call") # The logs are saved in the logs file. study.optimize(objective, n_trials=10) optuna.logging.disable_propagation() # Stop propagating logs to the root logger. logger.info("Logs from second optimize call") # The new logs for second optimize call are not saved. study.optimize(objective, n_trials=10) with open("foo.log") as f: assert f.readline().startswith("A new study created") assert f.readline() == "Logs from first optimize call\n" # Check for logs after second optimize call. assert f.read().split("Logs from second optimize call\n")[-1] == "" FNr0r&r-rrrdisable_propagationrEs&b#$$$+0(((rcHtdt_dS)aEnable propagation of the library log outputs. Please disable the Optuna's default handler to prevent double logging if the root logger has been configured. Example: Propagate all log output to the root logger in order to save them to the file. .. testsetup:: def objective(trial): x = trial.suggest_float("x", -100, 100) y = trial.suggest_categorical("y", [-1, 0, 1]) return x**2 + y .. testcode:: import optuna import logging logger = logging.getLogger() logger.setLevel(logging.INFO) # Setup the root logger. logger.addHandler(logging.FileHandler("foo.log", mode="w")) optuna.logging.enable_propagation() # Propagate logs to the root logger. optuna.logging.disable_default_handler() # Stop showing logs in sys.stderr. study = optuna.create_study() logger.info("Start optimization.") study.optimize(objective, n_trials=10) with open("foo.log") as f: assert f.readline().startswith("A new study created") assert f.readline() == "Start optimization.\n" TNrDrrrenable_propagationrG-s&R#$$$+/(((r)r r)r r)r r!)r r')r5rr r!)r r8)r<r8r r') __future__rr$rrrrrr r r threadingr__all__Lockr __annotations__r rr r&r0r4r7r;r>r@rBrErGrrrrMs"""""""    ' ((((((+/////    """"2222...."     ####3:3:3:3:l03030303f$?$?$?$?N < < < <21212121j*0*0*0*0*0*0r