install torch first

mehta-lab · Feb 24, 2024 · b0fdf77 · b0fdf77
1 parent 0e539af
commit b0fdf77
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/pr.yml b/.github/workflows/pr.yml
@@ -32,6 +32,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           # Install cpu wheels only to speed up the build
-          pip install ".[metrics,dev]" --index-url https://download.pytorch.org/whl/cpu
+          pip install torch --index-url https://download.pytorch.org/whl/cpu
+          pip install ".[metrics,dev]"
       - name: Test with pytest
         run: pytest -v