| # SPDX-License-Identifier: Apache-2.0 | |
| # adapted from vllm: https://github.com/vllm-project/vllm/blob/v0.7.3/vllm/logging_utils/formatter.py | |
| import logging | |
| class NewLineFormatter(logging.Formatter): | |
| """Adds logging prefix to newlines to align multi-line messages.""" | |
| def __init__(self, fmt, datefmt=None, style="%"): | |
| logging.Formatter.__init__(self, fmt, datefmt, style) | |
| def format(self, record): | |
| msg = logging.Formatter.format(self, record) | |
| if record.message != "": | |
| parts = msg.split(record.message) | |
| msg = msg.replace("\n", "\r\n" + parts[0]) | |
| return msg | |
| def setup_for_distributed(is_master): | |
| """ | |
| This function disables printing when not in master process | |
| """ | |
| import builtins as __builtin__ | |
| builtin_print = __builtin__.print | |
| def print(*args, **kwargs): | |
| force = kwargs.pop('force', False) | |
| if is_master or force: | |
| builtin_print(*args, **kwargs) | |
| __builtin__.print = print | |