Merge pull request #131 from johnwmillr/master

Fix tuple bug in output directory name
This commit is contained in:
Félix Voituret
2019-11-25 15:26:49 +01:00
committed by GitHub
2 changed files with 3 additions and 3 deletions

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(

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