Update torch dependency in tests

pyt-team · Jul 8, 2024 · 5973e04 · 5973e04
1 parent 04d9437
commit 5973e04
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/.github/workflows/test_codebase.yml b/.github/workflows/test_codebase.yml
@@ -28,9 +28,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ["3.10", "3.11"]
-        torch-version: [2.0.1]
-        include:
-          - torch-version: 2.0.1
+        torch-version: ["2.2", "2.3"]
 
     steps:
       - uses: actions/checkout@v3

diff --git a/.github/workflows/test_tutorials.yml b/.github/workflows/test_tutorials.yml
@@ -32,9 +32,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ["3.10", "3.11"]
-        torch-version: [2.0.1]
-        include:
-          - torch-version: 2.0.1
+        torch-version: ["2.2", "2.3"]
         test-group: [1, 2, 3, 4]
 
     steps: