From 2b18451eec5a50df8dbcf6ac27a4b766add15abf Mon Sep 17 00:00:00 2001 From: johnwmillr Date: Sat, 23 Nov 2019 15:42:40 -0800 Subject: [PATCH] Fix file naming tuple bug --- spleeter/separator.py | 2 +- tests/test_separator.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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')),