From 17748a8309f87c1fb51c5cec505a591f1e5fdefe Mon Sep 17 00:00:00 2001
From: Justin Kerr <justin.g.kerr@gmail.com>
Date: Wed, 31 Jan 2024 11:34:00 -0800
Subject: [PATCH] limit torch to <=2.1 to avoid type errors (#2857)

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

diff --git a/pyproject.toml b/pyproject.toml
index c4ea5fc15..f0b90702e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -53,7 +53,8 @@ dependencies = [
     "scikit-image>=0.19.3",
     "splines==0.3.0",
     "tensorboard>=2.13.0",
-    "torch>=1.13.1",
+    # pin torch to <=2.1 to fix https://github.com/pytorch/pytorch/issues/118736
+    "torch>=1.13.1,<=2.1",
     "torchvision>=0.14.1",
     "torchmetrics[image]>=1.0.1",
     "typing_extensions>=4.4.0",