yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-06 04:32:43 +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,
sample_rate=self._sample_rate)
sources = self.separate(waveform)
filename = splitext(basename(audio_descriptor))
filename = splitext(basename(audio_descriptor))[0]
generated = []
for instrument, data in sources.items():
path = join(destination, filename_format.format(

4
tests/test_separator.py
View File

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