From 1665861a766cd3a97d90062e3fdce6a40a104abf Mon Sep 17 00:00:00 2001 From: Faylixe Date: Tue, 8 Dec 2020 14:40:03 +0100 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20=20setup=20tensorflow=20logging=20h?= =?UTF-8?q?andler?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- spleeter/utils/logging.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/spleeter/utils/logging.py b/spleeter/utils/logging.py index bb1df18..5e0d557 100644 --- a/spleeter/utils/logging.py +++ b/spleeter/utils/logging.py @@ -10,7 +10,7 @@ from os import environ # pyright: reportMissingImports=false # pylint: disable=import-error -from tensorflow.compat.v1 import logging as tflogging +from tensorflow.compat.v1 import logging as tf_logging from typer import echo # pylint: enable=import-error @@ -42,11 +42,13 @@ def configure_logger(verbose: bool) -> None: verbose (bool): `True` to use verbose logger, `False` otherwise. """ + tf_logger = tf_logging._get_logger() + tf_logger.handlers = [handler] if verbose: environ['TF_CPP_MIN_LOG_LEVEL'] = '1' - tflogging.set_verbosity(tflogging.INFO) + tf_logging.set_verbosity(tf_logging.INFO) logger.setLevel(logging.DEBUG) else: warnings.filterwarnings('ignore') environ['TF_CPP_MIN_LOG_LEVEL'] = '3' - tflogging.set_verbosity(tflogging.ERROR) + tf_logging.set_verbosity(tf_logging.ERROR)