From e0ff2272297e9f179b883d8c2d6b05c6f33740ac Mon Sep 17 00:00:00 2001
From: Eitan Turok <150733043+eitanturok@users.noreply.github.com>
Date: Fri, 9 Aug 2024 16:50:32 -0400
Subject: [PATCH] bump torch to <2.5

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index fa15ee4..bdc76dc 100644
--- a/setup.py
+++ b/setup.py
@@ -62,7 +62,7 @@
 install_requires = [
     'numpy>=1.21.5,<2.1.0',
     'packaging>=21.3.0,<24.2',
-    'torch>=2.3.0,<2.4',
+    'torch>=2.3.0,<2.5',
     'triton>=2.1.0',
     'stanford-stk @ git+https://git@github.com/stanford-futuredata/stk.git@a1ddf98466730b88a2988860a9d8000fd1833301',
 ]