fixed an import exception

CuiDachao · Aug 23, 2022 · 19230dc · 19230dc
1 parent 39beb29
commit 19230dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/omegafold/pipeline.py b/omegafold/pipeline.py
@@ -43,7 +43,7 @@
 
 try:
     from torch.backends import mps  # Compatibility with earlier versions
-except IndexError:
+except ImportError:
     mps = None