Skip to content

Commit 8b6900c

Browse files
committed
Skip tests if unauthorised for model
1 parent f263900 commit 8b6900c

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

tests/test_mlip_calculators.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,14 +64,19 @@
6464

6565
try:
6666
from fairchem.core import pretrained_mlip
67+
from huggingface_hub.errors import GatedRepoError
6768

68-
UMA_PREDICT_UNIT = pretrained_mlip.get_predict_unit(
69-
model_name=UMA_LABEL, device="cpu"
70-
)
69+
try:
70+
UMA_PREDICT_UNIT = pretrained_mlip.get_predict_unit(
71+
model_name=UMA_LABEL, device="cpu"
72+
)
73+
except GatedRepoError:
74+
UMA_PREDICT_UNIT = None
7175

7276
except ImportError:
7377
UMA_PREDICT_UNIT = None
7478

79+
7580
ALIGNN_PATH = MODEL_PATH / "v5.27.2024"
7681
M3GNET_DIR_PATH = MODEL_PATH / "M3GNet-MP-2021.2.8-DIRECT-PES"
7782
M3GNET_MODEL_PATH = M3GNET_DIR_PATH / "model.pt"

0 commit comments

Comments
 (0)