fix: restrain PyTorch version to below 2.5 (#244)

matthieu-d4r · web-flow · commit 9a0c89f4112f · 2024-10-22T10:59:51.000+01:00
fix: restrain PyTorch version to below 2.5 Restrain PyTorch version to below 2.5, since the latter creates some problem in our CI and wheels building. See also: pytorch/pytorch#138333
diff --git a/s3torchbenchmarking/pyproject.toml b/s3torchbenchmarking/pyproject.toml
@@ -9,8 +9,9 @@ description = "Tools to run and compare benchmarks against various PyTorch conne
 requires-python = ">=3.8,<3.13"
 readme = "README.md"
 dependencies = [
-    #TODO: Remove torch != 2.3.0 restriction when https://github.com/pytorch/data/issues/1244 is fixed
-    "torch >= 2.0.1, != 2.3.0",
+    # TODO: remove "!= 2.3.0" restriction once https://github.com/pytorch/data/issues/1244 is fixed
+    # TODO: remove "< 2.5" restriction once https://github.com/pytorch/pytorch/issues/138333 is fixed
+    "torch >= 2.0.1, != 2.3.0, < 2.5",
     "lightning >= 2.0",
     "s3torchconnector",
     "hydra-core",
diff --git a/s3torchconnector/pyproject.toml b/s3torchconnector/pyproject.toml
@@ -22,7 +22,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "torch >= 2.0.1",
+    "torch >= 2.0.1, < 2.5", # TODO: remove "< 2.5" restriction once https://github.com/pytorch/pytorch/issues/138333 is fixed
     "s3torchconnectorclient >= 1.2.6",
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ classifiers = [`
`22`	`22`	`]`
`23`	`23`
`24`	`24`	`dependencies = [`
`25`		`- "torch >= 2.0.1",`
	`25`	`+ "torch >= 2.0.1, < 2.5", # TODO: remove "< 2.5" restriction once https://github.com/pytorch/pytorch/issues/138333 is fixed`
`26`	`26`	`"s3torchconnectorclient >= 1.2.6",`
`27`	`27`	`]`
`28`	`28`