attempt fix: torch installations

arangoml · Oct 6, 2023 · 4925a49 · 4925a49
1 parent 26125ad
commit 4925a49
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -25,8 +25,8 @@ jobs:
           cache-dependency-path: setup.py
       - name: Install packages
         run: |
-          pip install torch
-          pip install torch-sparse
+          pip install torch==2.1.0
+          pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.1.0+cpu.html
           pip install -e .[dev]
       - name: Run black
         run: black --check --verbose --diff --color ${{env.PACKAGE_DIR}} ${{env.TESTS_DIR}}