tests

mlmed · Dec 12, 2023 · c15277e · c15277e
1 parent 15e7d3f
commit c15277e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -32,7 +32,7 @@ jobs:
       max-parallel: 2
       matrix:
         python-version: ['3.9']
-        torch-version: ['--upgrade']
+        torch-version: [2.1.1]
         os: [ubuntu-latest] # only run ubuntu for now because the other ones fail for no reason, macos-latest, windows-latest]
 
     # Steps represent a sequence of tasks that will be executed as part of the job