🐛 fix path, entrypoint and logging

This commit is contained in:
Faylixe
2020-12-11 12:48:28 +01:00
parent 91c9691c30
commit 9e45cd0352
2 changed files with 11 additions and 7 deletions

View File

@@ -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()

View File

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