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'))