mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-30 12:22:58 +00:00
🐛 fix evaluation test
This commit is contained in:
@@ -72,10 +72,13 @@ def test_evaluate(backend):
|
||||
with TemporaryDirectory() as directory:
|
||||
generate_fake_eval_dataset(directory)
|
||||
metrics = evaluate(
|
||||
adapter='spleeter.audio.adapter.ffmpeg.FFMPEGProcessAudioAdapter',
|
||||
output_path='eval',
|
||||
stft_backend=backend,
|
||||
params_filename='spleeter:4stems',
|
||||
mus_dir=directory,
|
||||
)
|
||||
mwf=False,
|
||||
verbose=False)
|
||||
for instrument, metric in metrics.items():
|
||||
for m, value in metric.items():
|
||||
assert np.allclose(
|
||||
|
||||
@@ -101,7 +101,6 @@ def test_train():
|
||||
'-p', 'useless_config.json',
|
||||
'-d', path
|
||||
])
|
||||
raise IOError(result.stdout)
|
||||
assert result.exit_code == 0
|
||||
# assert that model checkpoint was created.
|
||||
assert os.path.exists(join(path, 'model', 'model.ckpt-10.index'))
|
||||
|
||||
Reference in New Issue
Block a user