From 981efbd7341b722b77de5f839e6a9906d9c34118 Mon Sep 17 00:00:00 2001 From: Faylixe Date: Tue, 8 Dec 2020 14:53:07 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20=20fix=20test?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- spleeter/utils/logging.py | 4 ++-- tests/test_train.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/spleeter/utils/logging.py b/spleeter/utils/logging.py index 5e0d557..b97349d 100644 --- a/spleeter/utils/logging.py +++ b/spleeter/utils/logging.py @@ -42,8 +42,8 @@ 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] + # tf_logger = tf_logging._get_logger() + # tf_logger.handlers = [handler] if verbose: environ['TF_CPP_MIN_LOG_LEVEL'] = '1' tf_logging.set_verbosity(tf_logging.INFO) diff --git a/tests/test_train.py b/tests/test_train.py index dd6f306..d5947c2 100644 --- a/tests/test_train.py +++ b/tests/test_train.py @@ -101,7 +101,7 @@ def test_train(): '-p', 'useless_config.json', '-d', path ]) - raise IOError(f'STDOUT: {result.stdout}, STDERR: {result.stderr}') + raise IOError(f'STDOUT: {result.stdout}') assert result.exit_code == 0 # assert that model checkpoint was created. assert os.path.exists(join(path, 'model', 'model.ckpt-10.index'))