Attempt to fix torch version

microsoft · Aug 6, 2024 · 947ca7c · 947ca7c
1 parent 77071f2
commit 947ca7c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -25,7 +25,6 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install torch==2.2.0  # We require this specific version for numerical tests!
           python -m pip install --upgrade --no-cache-dir -e '.[dev]'
       - name: Log into HuggingFace
         run: |

diff --git a/pyproject.toml b/pyproject.toml
@@ -50,6 +50,7 @@ dev = [
     "ruff==0.4.1",
     "pre-commit",
     "jupyter-book",
+    "torch==2.2.0",
 ]
 
 [project.urls]