🐛 fix adapter option

This commit is contained in:
Faylixe
2020-12-08 15:10:32 +01:00
parent 981efbd734
commit 914f4e01ef
3 changed files with 3 additions and 4 deletions

View File

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

View File

@@ -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',

View File

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