diff --git a/s3torchbenchmarking/pyproject.toml b/s3torchbenchmarking/pyproject.toml index 912d284a..b331e503 100644 --- 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 index d8e8941f..b08b8c14 100644 --- 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", ]