update requirements and pytorch version

deel-ai · Oct 14, 2024 · 2cfa164 · 2cfa164
1 parent 9696ab4
commit 2cfa164
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 9 deletions.
diff --git a/.github/workflows/python-tests.yml b/.github/workflows/python-tests.yml
@@ -10,7 +10,7 @@ jobs:
       matrix:
         include:
           - python-version: 3.9
-            pt-version: 1.10.2
+            pt-version: 1.13.1
           - python-version: "3.10"
             pt-version: 2.1.2
           - python-version: "3.11"

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,2 @@
-torch_testing==0.0.2
-inflection==0.5.1
-numpy
-torch==1.10.2+cu113
-torchvision==0.11.3+cu113
+numpy<2.0
+torch>=1.13.1
diff --git a/setup.cfg b/setup.cfg
@@ -7,15 +7,13 @@ extend-ignore =
 
 [tox:tox]
 envlist = 
-    py{39,310,311}-pt{1.10.2,1.13.1,2.1.2,2.3.1,2.4.1}
+    py{39,310,311}-pt{1.13.1,2.1.2,2.3.1,2.4.1}
     py{39,310,311}-lint
 
 [testenv]
 pip_version = pip>=20
 deps =
     pytest
-    pt1.10.2: torch==1.10.2+cu113
-    pt1.10.2: numpy<2.0
     pt1.13.1: torch==1.13.1+cu117
     pt1.13.1: numpy<2.0
     pt2.1.2: torch==2.1.2+cu121