yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-06 04:32:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix enum check

Browse Source
This commit is contained in:
Faylixe
2020-12-08 13:34:16 +01:00
parent 72e74aaf22
commit a4d668b612
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
spleeter/audio/__init__.py
View File

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