From 9e45cd0352ac6ed6499314548f78048c8c3ad63f Mon Sep 17 00:00:00 2001 From: Faylixe Date: Fri, 11 Dec 2020 12:48:28 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20=20fix=20path,=20entrypoint=20an?= =?UTF-8?q?d=20logging?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- spleeter/__main__.py | 11 ++++++++--- spleeter/utils/logging.py | 7 +++---- 2 files changed, 11 insertions(+), 7 deletions(-) diff --git a/spleeter/__main__.py b/spleeter/__main__.py index d2ca490..a2e671f 100644 --- a/spleeter/__main__.py +++ b/spleeter/__main__.py @@ -112,8 +112,8 @@ def separate( stft_backend=stft_backend) for filename in files: separator.separate_to_file( - filename, - output_path, + str(filename), + str(output_path), audio_adapter=audio_adapter, offset=offset, duration=duration, @@ -221,8 +221,13 @@ def evaluate( return metrics -if __name__ == '__main__': +def entrypoint(): + """ Application entrypoint. """ try: spleeter() except SpleeterError as e: logger.error(e) + + +if __name__ == '__main__': + entrypoint() diff --git a/spleeter/utils/logging.py b/spleeter/utils/logging.py index 1db3bd6..fd6ae72 100644 --- a/spleeter/utils/logging.py +++ b/spleeter/utils/logging.py @@ -10,9 +10,6 @@ from os import environ # pyright: reportMissingImports=false # pylint: disable=import-error -import tensorflow as tf - -from tensorflow.compat.v1 import logging as tf_logging from typer import echo # pylint: enable=import-error @@ -46,7 +43,9 @@ def configure_logger(verbose: bool) -> None: verbose (bool): `True` to use verbose logger, `False` otherwise. """ - tf_logger = tf.get_logger() + from tensorflow import get_logger + from tensorflow.compat.v1 import logging as tf_logging + tf_logger = get_logger() tf_logger.handlers = [handler] if verbose: tf_logging.set_verbosity(tf_logging.INFO)