From cb00099e10034a477b2e6bf42587f3884b259ac2 Mon Sep 17 00:00:00 2001 From: dnnanuti Date: Tue, 30 Apr 2024 14:01:46 +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 | 2 +- s3torchconnector/pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/s3torchbenchmarking/pyproject.toml b/s3torchbenchmarking/pyproject.toml index 912d284a..6f8ca9ca 100644 --- 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 index d8e8941f..eadc2a09 100644 --- 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", ]