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