Restrict torch version to be different than 2.3.0 due to:

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

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/s3torchbenchmarking/pyproject.toml b/s3torchbenchmarking/pyproject.toml
@@ -9,7 +9,7 @@ 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",
+    "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,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "torch >= 2.0.1",
+    "torch >= 2.0.1, !=2.3.0",
     "s3torchconnectorclient >= 1.2.3",
 ]