Skip to content

Commit 897a4f4

Browse files
committed
try loading UMA model in CI (requires huggingface auth)
using my personal HF token for now
1 parent 231dcfc commit 897a4f4

File tree

1 file changed

+7
-24
lines changed

1 file changed

+7
-24
lines changed

.github/workflows/test.yml

Lines changed: 7 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -82,14 +82,6 @@ jobs:
8282
- name: Check out repo
8383
uses: actions/checkout@v4
8484

85-
- name: Check out fairchem repository
86-
if: ${{ matrix.model.name == 'fairchem' }}
87-
uses: actions/checkout@v4
88-
with:
89-
repository: FAIR-Chem/fairchem
90-
path: fairchem-repo
91-
ref: fairchem_core-1.10.0
92-
9385
- name: Set up Python
9486
uses: actions/setup-python@v5
9587
with:
@@ -98,24 +90,13 @@ jobs:
9890
- name: Set up uv
9991
uses: astral-sh/setup-uv@v6
10092

101-
- name: Install fairchem repository and dependencies
93+
- name: Install fairchem and dependencies
10294
if: ${{ matrix.model.name == 'fairchem' }}
103-
env:
104-
HF_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
10595
run: |
106-
uv pip install huggingface_hub --system
107-
if [ -n "$HF_TOKEN" ]; then
108-
huggingface-cli login --token "$HF_TOKEN"
109-
else
110-
echo "HF_TOKEN is not set. Skipping login."
111-
fi
112-
if [ -f fairchem-repo/packages/requirements.txt ]; then
113-
uv pip install -r fairchem-repo/packages/requirements.txt --system
114-
fi
115-
if [ -f fairchem-repo/packages/requirements-optional.txt ]; then
116-
uv pip install -r fairchem-repo/packages/requirements-optional.txt --system
117-
fi
118-
uv pip install -e fairchem-repo/packages/fairchem-core[dev] --system
96+
uv pip install "torch>2" --index-url https://download.pytorch.org/whl/cpu --system
97+
uv pip install "torch-scatter" -f https://data.pyg.org/whl/torch-2.6.0+cpu.html --system
98+
uv pip install "torch-sparse" -f https://data.pyg.org/whl/torch-2.6.0+cpu.html --system
99+
uv pip install "fairchem-core>=2.2.0" --system
119100
uv pip install -e .[test] --resolution=${{ matrix.version.resolution }} --system
120101
121102
- name: Install torch_sim with model dependencies
@@ -124,6 +105,8 @@ jobs:
124105
uv pip install -e .[test,${{ matrix.model.name }}] --resolution=${{ matrix.version.resolution }} --system
125106
126107
- name: Run ${{ matrix.model.test_path }} tests
108+
env:
109+
HF_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
127110
run: |
128111
pytest --cov=torch_sim --cov-report=xml ${{ matrix.model.test_path }}
129112

0 commit comments

Comments
 (0)