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 file naming tuple bug

Browse Source
This commit is contained in:
johnwmillr
2019-11-23 15:42:40 -08:00
parent 85ff00797f
commit 2b18451eec
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
spleeter/separator.py
View File

@@ -121,7 +121,7 @@ class Separator(object):
duration=duration, duration=duration,
sample_rate=self._sample_rate) sample_rate=self._sample_rate)
sources = self.separate(waveform) sources = self.separate(waveform)
filename = splitext(basename(audio_descriptor)) filename = splitext(basename(audio_descriptor))[0]
generated = [] generated = []
for instrument, data in sources.items(): for instrument, data in sources.items():
path = join(destination, filename_format.format( path = join(destination, filename_format.format(

4
tests/test_separator.py
View File

@@ -9,7 +9,7 @@ __license__ = 'MIT License'
import filecmp import filecmp
from os.path import splitext, exists, join from os.path import splitext, basename, exists, join
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
import pytest import pytest
@@ -19,7 +19,7 @@ from spleeter.audio.adapter import get_default_audio_adapter
from spleeter.separator import Separator from spleeter.separator import Separator
TEST_AUDIO_DESCRIPTOR = 'audio_example.mp3' TEST_AUDIO_DESCRIPTOR = 'audio_example.mp3'
TEST_AUDIO_BASENAME = splitext(TEST_AUDIO_DESCRIPTOR) TEST_AUDIO_BASENAME = splitext(basename(TEST_AUDIO_DESCRIPTOR))[0]
TEST_CONFIGURATIONS = [ TEST_CONFIGURATIONS = [
('spleeter:2stems', ('vocals', 'accompaniment')), ('spleeter:2stems', ('vocals', 'accompaniment')),
('spleeter:4stems', ('vocals', 'drums', 'bass', 'other')), ('spleeter:4stems', ('vocals', 'drums', 'bass', 'other')),