From 914f4e01efdd872966fb070317c65c0d6c3a1384 Mon Sep 17 00:00:00 2001 From: Faylixe Date: Tue, 8 Dec 2020 15:10:32 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20=20fix=20adapter=20option?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- spleeter/options.py | 2 +- tests/test_eval.py | 2 +- tests/test_train.py | 3 +-- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/spleeter/options.py b/spleeter/options.py index 99cf1ad..d5c89db 100644 --- a/spleeter/options.py +++ b/spleeter/options.py @@ -26,7 +26,7 @@ AudioInputArgument: ArgumentInfo = Argument( resolve_path=True) AudioAdapterOption: OptionInfo = Option( - 'spleeter.audio.adapter.ffmpeg.FFMPEGProcessAudioAdapter', + 'spleeter.audio.ffmpeg.FFMPEGProcessAudioAdapter', '--adapter', '-a', help='Name of the audio adapter to use for audio I/O') diff --git a/tests/test_eval.py b/tests/test_eval.py index 4103cad..4f90cac 100644 --- a/tests/test_eval.py +++ b/tests/test_eval.py @@ -72,7 +72,7 @@ def test_evaluate(backend): with TemporaryDirectory() as directory: generate_fake_eval_dataset(directory) metrics = evaluate( - adapter='spleeter.audio.adapter.ffmpeg.FFMPEGProcessAudioAdapter', + adapter='spleeter.audio.ffmpeg.FFMPEGProcessAudioAdapter', output_path='eval', stft_backend=backend, params_filename='spleeter:4stems', diff --git a/tests/test_train.py b/tests/test_train.py index d5947c2..47ce747 100644 --- a/tests/test_train.py +++ b/tests/test_train.py @@ -101,9 +101,8 @@ def test_train(): '-p', 'useless_config.json', '-d', path ]) - 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')) assert os.path.exists(join(path, 'model', 'checkpoint')) assert os.path.exists(join(path, 'model', 'model.ckpt-0.meta')) + assert result.exit_code == 0