Restrict torch version to be different than 2.3.0 due to:

pytorch/data#1244
awslabs · Apr 30, 2024 · 21c8cfc · 21c8cfc
1 parent 06c1d56
commit 21c8cfc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/s3torchbenchmarking/pyproject.toml b/s3torchbenchmarking/pyproject.toml
@@ -9,7 +9,8 @@ description = "Tools to run and compare benchmarks against various PyTorch conne
 requires-python = ">=3.8,<3.13"
 readme = "README.md"
 dependencies = [
-    "torch >= 2.0.1",
+    #TODO: Remove torch != 2.3.0 restriction when https://github.com/pytorch/data/issues/1244 is fixed
+    "torch >= 2.0.1, != 2.3.0",
     "s3torchconnector",
     "hydra-core",
     "torchdata>=0.6.1",

diff --git a/s3torchconnector/pyproject.toml b/s3torchconnector/pyproject.toml
@@ -22,7 +22,8 @@ classifiers = [
 ]
 
 dependencies = [
-    "torch >= 2.0.1",
+    #TODO: Remove torch != 2.3.0 restriction when https://github.com/pytorch/data/issues/1244 is fixed
+    "torch >= 2.0.1, != 2.3.0",
     "s3torchconnectorclient >= 1.2.3",
 ]