From a4d668b612bba426fe2f32aca2811c5f7240fcc5 Mon Sep 17 00:00:00 2001 From: Faylixe Date: Tue, 8 Dec 2020 13:34:16 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20=20fix=20enum=20check?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- spleeter/audio/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/spleeter/audio/__init__.py b/spleeter/audio/__init__.py index f3b15e2..c704aa1 100644 --- a/spleeter/audio/__init__.py +++ b/spleeter/audio/__init__.py @@ -42,8 +42,7 @@ class STFTBackend(str, Enum): @classmethod def resolve(cls: type, backend: str) -> str: - backend = backend.upper() - if backend not in cls.__members__.keys(): + if backend not in cls.__members__.values(): raise ValueError(f'Unsupported backend {backend}') if backend == cls.AUTO: if len(tf.config.list_physical_devices('GPU')):