From 1436869963066caa06d0d44303ee894b7e68cda3 Mon Sep 17 00:00:00 2001
From: Gautam <gautam@aperturedata.io>
Date: Thu, 29 Aug 2024 12:08:54 -0400
Subject: [PATCH] Increasing the window of supporting pytorch version to
 current stable, i.e 2.4.0, and torchvision to 0.19.9

Removed a redundant entry in requirements.

Ran the tests locally with a new virtual env.
---
 setup.py                       | 4 ++--
 tests/actions_requirements.txt | 2 --
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index 1864989b..fd4e44d0 100644
--- a/setup.py
+++ b/setup.py
@@ -39,8 +39,8 @@
         'numpy>=1.24.0,<2.0.0',
         'Pillow>=10.2.0,<10.3.0',
         'requests>=2.0.0,<3.0.0',
-        'torch>=2.2.0,<=2.3.0',
-        'torchvision>=0.17.0,<=0.18.0',
+        'torch>=2.2.0,<=2.4.0',
+        'torchvision>=0.17.0,<=0.19.0',
         'tqdm>=4.0.0,<5.0.0',
     ],
 )
diff --git a/tests/actions_requirements.txt b/tests/actions_requirements.txt
index b74924a5..30e699e9 100644
--- a/tests/actions_requirements.txt
+++ b/tests/actions_requirements.txt
@@ -1,7 +1,5 @@
 numpy>=1.24.0,<2.0.0
 requests>=2.0.0,<3.0.0
-torch>=2.2.0,<2.3.0
-torchvision>=0.17.0,<0.18.0
 Pillow>=10.2.0,<10.3.0
 opencv-python>=4.9.0
 scipy>=1.10.0,<2.0.0