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