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 f263900 commit 8b6900cCopy full SHA for 8b6900c
tests/test_mlip_calculators.py
@@ -64,14 +64,19 @@
64
65
try:
66
from fairchem.core import pretrained_mlip
67
+ from huggingface_hub.errors import GatedRepoError
68
- UMA_PREDICT_UNIT = pretrained_mlip.get_predict_unit(
69
- model_name=UMA_LABEL, device="cpu"
70
- )
+ try:
+ UMA_PREDICT_UNIT = pretrained_mlip.get_predict_unit(
71
+ model_name=UMA_LABEL, device="cpu"
72
+ )
73
+ except GatedRepoError:
74
+ UMA_PREDICT_UNIT = None
75
76
except ImportError:
77
UMA_PREDICT_UNIT = None
78
79
+
80
ALIGNN_PATH = MODEL_PATH / "v5.27.2024"
81
M3GNET_DIR_PATH = MODEL_PATH / "M3GNet-MP-2021.2.8-DIRECT-PES"
82
M3GNET_MODEL_PATH = M3GNET_DIR_PATH / "model.pt"
0 commit comments