o 㥵i @s>ddlZddlmZmZddlmZmZGdddejZdS)N)MappingOptional)rank_prefixed_messagerank_zero_onlyc sheZdZdZeddfdededeeeefddffdd Z dd e d ed ee ddfd dZ Z S) RankedLoggerz0A multi-GPU-friendly python command line logger.TNnamerextrareturncs$t|}tj||d||_dS)aInitializes a multi-GPU-friendly python command line logger that logs on all processes with their rank prefixed in the log message. :param name: The name of the logger. Default is ``__name__``. :param rank_zero_only: Whether to force all logs to only occur on the rank zero process. Default is `False`. :param extra: (Optional) A dict-like object which provides contextual information. See `logging.LoggerAdapter`. )loggerrN)logging getLoggersuper__init__r)selfrrrr  __class__L/home/ubuntu/.local/lib/python3.10/site-packages/fish_speech/utils/logger.pyr s  zRankedLogger.__init__levelmsgrankcOs||ra|||\}}ttdd}|durtdt||}|jr9|dkr7|jj||g|Ri|dSdS|durM|jj||g|Ri|dS||krc|jj||g|Ri|dSdSdS)aDDelegate a log call to the underlying logger, after prefixing its message with the rank of the process it's being logged from. If `'rank'` is provided, then the log will only occur on that rank/process. :param level: The level to log at. Look at `logging.__init__.py` for more information. :param msg: The message to log. :param rank: The rank to log at. :param args: Additional args to pass to the underlying logging function. :param kwargs: Any additional keyword args to pass to the underlying logging function. rNz4The `rank_zero_only.rank` needs to be set before user) isEnabledForprocessgetattrr RuntimeErrorrr log)rrrrargskwargs current_rankrrrrs$      zRankedLogger.log)N) __name__ __module__ __qualname____doc__strboolrrobjectrintr __classcell__rrrrrs0r) r typingrr"lightning_utilities.core.rank_zerorr LoggerAdapterrrrrrs