lock torch in workflow

arangoml · Nov 14, 2023 · 98cdcd0 · 98cdcd0
1 parent 92ac4ea
commit 98cdcd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -32,7 +32,7 @@ jobs:
 
  - name: Install packages
  run: |
- pip install torch
+ pip install torch==2.1.0
  pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-$(python -c 'import torch; print(torch.__version__.split("+")[0])')+cpu.html
  pip install -e .[dev]