o ̳iB @sddlZddlZddlmZmZddlmZmZmZddl m Z ede dZ ddeedejfd d Zedejfd ejd ededdfd dZdd edee ge ffddZejfd ejd ededdfddZdS)N) lru_cachewraps)CallableOptionalTypeVar) distributedT)boundlevelreturncCsDtt}|s|t|dur tt|}|||S)a{ Get a logger with a stream handler. Args: level (Optional[str]): The logging level. See https://docs.python.org/3/library/logging.html#levels for list of levels. Example: >>> logger = get_logger("INFO") >>> logger.info("Hello world!") INFO:torchtune.utils._logging:Hello world! Returns: logging.Logger: The logger. N) logging getLogger__name__ hasHandlers addHandler StreamHandlergetattruppersetLevel)r loggerrL/home/ubuntu/.local/lib/python3.10/site-packages/torchtune/utils/_logging.py get_loggers  rrmsgcCst|||ddS)a Logs a message only once. LRU cache is used to ensure a specific message is logged only once, similar to how :func:`~warnings.warn` works when the ``once`` rule is set via command-line or environment variable. Args: logger (logging.Logger): The logger. msg (str): The warning message. level (int): The logging level. See https://docs.python.org/3/library/logging.html#levels for values. Defaults to ``logging.INFO``. rrr N) log_rank_zerorrrrlog_once)s rcs&tddfddfdd}|S)z Decorator to mark an object as deprecated and print additional message. Args: msg (str): additional information to print after warning. Returns: Callable[[T], T]: the decorated object. )maxsizecsHtr tr tnd}|dkrdStj|jdtdddS)Nrz7 is deprecated and will be removed in future versions. )category stacklevel)dist is_availableis_initializedget_rankwarningswarnr FutureWarning)objrank)rrrr(Ds  zdeprecated..warncstfdd}|S)Ncs|i|SNr)argskwargs)r*r(rrwrapperQsz.deprecated..decorator..wrapper)r)r*r/)r()r*r decoratorPszdeprecated..decorator)r)rr0r)rr(r deprecated9s  r1cCs<tr tr tnd}|dkrdS|j||dddS)a# Logs a message only on rank zero. Args: logger (logging.Logger): The logger. msg (str): The warning message. level (int): The logging level. See https://docs.python.org/3/library/logging.html#levels for values. Defaults to ``logging.INFO``. rN)r")r#r$r%r&log)rrr r+rrrr rr,)r)r r' functoolsrrtypingrrrtorchrr#typerstrLoggerrINFOintrr1rrrrrs  $&"