From 6436742c34bf3f8852b43581ea352ca15f95b5cd Mon Sep 17 00:00:00 2001
From: Brent Yi <yibrenth@gmail.com>
Date: Thu, 1 Feb 2024 16:19:54 -0800
Subject: [PATCH] Upper-bound torch version for dev installs only (#2864)

---
 pyproject.toml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 8b60c4bba..0b0a6c309 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -53,8 +53,7 @@ dependencies = [
     "scikit-image>=0.19.3",
     "splines==0.3.0",
     "tensorboard>=2.13.0",
-    # pin torch to <=2.1 to fix https://github.com/pytorch/pytorch/issues/118736
-    "torch>=1.13.1,<2.2",
+    "torch>=1.13.1",
     "torchvision>=0.14.1",
     "torchmetrics[image]>=1.0.1",
     "typing_extensions>=4.4.0",
@@ -102,6 +101,8 @@ dev = [
     # NOTE: Disabling projectaria-tools because it doesn't have prebuilt windows wheels
     # Syntax comes from here: https://pip.pypa.io/en/stable/reference/requirement-specifiers/
     "projectaria-tools>=1.3.1; sys_platform != 'win32'",
+    # pin torch to <=2.1 to fix https://github.com/pytorch/pytorch/issues/118736
+    "torch>=1.13.1,<2.2",
 ]
 
 # Documentation related packages