Fix tests?

metatensor · Feb 25, 2024 · 77d5848 · 77d5848
1 parent ff1377b
commit 77d5848
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/pet-jax.yml b/.github/workflows/pet-jax.yml
@@ -30,6 +30,9 @@ jobs:
 
     - name: run PET-JAX tests
       run: tox -e pet-jax-tests
+      env:
+        # Use the CPU only version of torch when building/running the code
+        PIP_EXTRA_INDEX_URL: https://download.pytorch.org/whl/cpu
 
     - name: Upload codecoverage
       uses: codecov/codecov-action@v4