yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-31 14:58:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix adapter option

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
spleeter/options.py
View File

@@ -26,7 +26,7 @@ AudioInputArgument: ArgumentInfo = Argument(
resolve_path=True) resolve_path=True)
AudioAdapterOption: OptionInfo = Option( AudioAdapterOption: OptionInfo = Option(
'spleeter.audio.adapter.ffmpeg.FFMPEGProcessAudioAdapter', 'spleeter.audio.ffmpeg.FFMPEGProcessAudioAdapter',
'--adapter', '--adapter',
'-a', '-a',
help='Name of the audio adapter to use for audio I/O') help='Name of the audio adapter to use for audio I/O')

2
tests/test_eval.py
View File

@@ -72,7 +72,7 @@ def test_evaluate(backend):
with TemporaryDirectory() as directory: with TemporaryDirectory() as directory:
generate_fake_eval_dataset(directory) generate_fake_eval_dataset(directory)
metrics = evaluate( metrics = evaluate(
adapter='spleeter.audio.adapter.ffmpeg.FFMPEGProcessAudioAdapter', adapter='spleeter.audio.ffmpeg.FFMPEGProcessAudioAdapter',
output_path='eval', output_path='eval',
stft_backend=backend, stft_backend=backend,
params_filename='spleeter:4stems', params_filename='spleeter:4stems',

3
tests/test_train.py
View File

@@ -101,9 +101,8 @@ def test_train():
'-p', 'useless_config.json', '-p', 'useless_config.json',
'-d', path '-d', path
]) ])
raise IOError(f'STDOUT: {result.stdout}')
assert result.exit_code == 0
# assert that model checkpoint was created. # assert that model checkpoint was created.
assert os.path.exists(join(path, 'model', 'model.ckpt-10.index')) 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', 'checkpoint'))
assert os.path.exists(join(path, 'model', 'model.ckpt-0.meta')) assert os.path.exists(join(path, 'model', 'model.ckpt-0.meta'))
assert result.exit_code == 0