From 21c8cfccbf9ecd264c630dfaab66a89efddbb254 Mon Sep 17 00:00:00 2001 From: dnnanuti Date: Tue, 30 Apr 2024 14:13:14 +0100 Subject: [PATCH] Restrict torch version to be different than 2.3.0 due to: https://github.com/pytorch/data/issues/1244 --- s3torchbenchmarking/pyproject.toml | 3 ++- s3torchconnector/pyproject.toml | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) 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", ]