We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4c11594 commit 89f764fCopy full SHA for 89f764f
tests/test_mlip_calculators.py
@@ -66,11 +66,14 @@
66
from fairchem.core import pretrained_mlip
67
from huggingface_hub.errors import GatedRepoError
68
69
- UMA_PREDICT_UNIT = pretrained_mlip.get_predict_unit(
70
- model_name=UMA_LABEL, device="cpu"
71
- )
+ try:
+ UMA_PREDICT_UNIT = pretrained_mlip.get_predict_unit(
+ model_name=UMA_LABEL, device="cpu"
72
+ )
73
+ except GatedRepoError:
74
+ UMA_PREDICT_UNIT = None
75
-except (ImportError, GatedRepoError):
76
+except ImportError:
77
UMA_PREDICT_UNIT = None
78
79
0 commit comments