fix(colab): color log printing (2noise#430)

waywayyang · Jun 24, 2024 · cae0fec · cae0fec
1 parent 25b6023
commit cae0fec
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/ipynb/colab.ipynb b/examples/ipynb/colab.ipynb
@@ -72,7 +72,7 @@
       },
       "outputs": [],
       "source": [
-        "logger = get_logger(\"ChatTTS\", remove_exist=True)\n",
+        "logger = get_logger(\"ChatTTS\", format_root=True)\n",
         "chat = ChatTTS.Chat(logger)\n",
         "\n",
         "# try to load normalizer\n",

diff --git a/tools/logger/log.py b/tools/logger/log.py
@@ -51,7 +51,7 @@ def format(self, record: logging.LogRecord):
         return logstr
 
 
-def get_logger(name: str, lv = logging.INFO, remove_exist=False):
+def get_logger(name: str, lv = logging.INFO, remove_exist=False, format_root=False):
     logger = logging.getLogger(name)
     logger.setLevel(lv)
     if remove_exist and logger.hasHandlers():
@@ -63,4 +63,7 @@ def get_logger(name: str, lv = logging.INFO, remove_exist=False):
     else:
         for h in logger.handlers:
             h.setFormatter(Formatter())
+    if format_root:
+        for h in logger.root.handlers:
+            h.setFormatter(Formatter())
     return logger