upgrade dependencies

awslabs · Jul 2, 2024 · d67fe07 · d67fe07
1 parent b7e5cb6
commit d67fe07
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/run_unit_tests.yml b/.github/workflows/run_unit_tests.yml
@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10", "3.11"]
+        python-version: ["3.8", "3.9", "3.10"]
 
     steps:
       - name: Checkout

diff --git a/requirements.txt b/requirements.txt
@@ -10,7 +10,7 @@ Pillow>=10.3.0, <10.4.1
 tabulate>=0.9.0, <0.9.1
 tensorboardX>=2.5.0, <2.6.3
 torchmetrics>=0.11.0, <0.11.5
-torchvision>=0.13.0, <0.18.2
+torchvision>=0.13.0, <0.16.3
 deepspeed>=0.9.0, <0.14.5
 datasets>=2.9.0, <2.20.1
 transformers>=4.38.0, <4.42.4