From 2cfa16436fe1de3c5affb21402bb1bd1d503bc92 Mon Sep 17 00:00:00 2001 From: Franck Mamalet Date: Mon, 14 Oct 2024 15:22:00 +0200 Subject: [PATCH] update requirements and pytorch version --- .github/workflows/python-tests.yml | 2 +- requirements.txt | 7 ++----- setup.cfg | 4 +--- 3 files changed, 4 insertions(+), 9 deletions(-) diff --git a/.github/workflows/python-tests.yml b/.github/workflows/python-tests.yml index d2f181f..548760b 100644 --- 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 index f6bd89a..9a04451 100644 --- 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 index 3907885..9f00238 100644 --- 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