Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(utils.generic_utils): add custom formatter for logging to console

This commit is contained in:
Enno Hermann 2024-04-02 12:15:18 +02:00
parent 9b2d48f8a6
commit ab64844aba
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
TTS/utils/generic_utils.py
View File

@ -129,6 +129,20 @@ def get_timestamp() -> str:
return datetime.now().strftime("%y%m%d-%H%M%S") return datetime.now().strftime("%y%m%d-%H%M%S")
class ConsoleFormatter(logging.Formatter):
"""Custom formatter that prints logging.INFO messages without the level name.
Source: https://stackoverflow.com/a/62488520
"""
def format(self, record):
if record.levelno == logging.INFO:
self._style._fmt = "%(message)s"
else:
self._style._fmt = "%(levelname)s: %(message)s"
return super().format(record)
def setup_logger( def setup_logger(
logger_name: str, logger_name: str,
level: int = logging.INFO, level: int = logging.INFO,