diff --git a/spleeter/separator.py b/spleeter/separator.py index 48d0b9d..c1c8f0e 100644 --- a/spleeter/separator.py +++ b/spleeter/separator.py @@ -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( diff --git a/tests/test_separator.py b/tests/test_separator.py index 43a1dda..9235731 100644 --- a/tests/test_separator.py +++ b/tests/test_separator.py @@ -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')),