From bed48b59120a868a153124d5e855fae4936b2755 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 20 Aug 2024 15:35:04 -0700 Subject: [PATCH] fix tests fix tests --- .github/workflows/python-test.yml | 1 + setup.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/python-test.yml b/.github/workflows/python-test.yml index 0da3b3b..fe54d29 100644 --- a/.github/workflows/python-test.yml +++ b/.github/workflows/python-test.yml @@ -28,6 +28,7 @@ jobs: python -m pip install --upgrade pip python -m pip install pytest python -m pip install wheel + python -m pip install torch==2.4.0 torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cpu if [ -f requirements.txt ]; then pip install -r requirements.txt; fi - name: Test with pytest run: | diff --git a/setup.py b/setup.py index 794ed22..aa47dfe 100644 --- a/setup.py +++ b/setup.py @@ -29,8 +29,8 @@ setup( ], tests_require=[ 'pytest', - 'torch==1.12.1', - 'torchvision==0.13.1' + 'torch==2.4.0', + 'torchvision==0.19.0' ], classifiers=[ 'Development Status :: 4 - Beta',