🐛 fix resolution

This commit is contained in:
Faylixe
2020-12-08 13:28:10 +01:00
parent 96901a8393
commit 72e74aaf22

View File

@@ -42,7 +42,8 @@ class STFTBackend(str, Enum):
@classmethod
def resolve(cls: type, backend: str) -> str:
if backend not in cls.__members__.items():
backend = backend.upper()
if backend not in cls.__members__.keys():
raise ValueError(f'Unsupported backend {backend}')
if backend == cls.AUTO:
if len(tf.config.list_physical_devices('GPU')):