Skip to content

Commit 18c7c12

Browse files
IsaevIlyaIlya Isaev
andauthored
Update torch version and bind torchdata to last version with DataPipes (#283)
Co-authored-by: Ilya Isaev <[email protected]>
1 parent a7667c1 commit 18c7c12

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

s3torchbenchmarking/pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,11 @@ readme = "README.md"
1111
dependencies = [
1212
# TODO: remove "!= 2.3.0" restriction once https://github.com/pytorch/data/issues/1244 is fixed
1313
# TODO: remove "< 2.5" restriction once https://github.com/pytorch/pytorch/issues/138333 is fixed
14-
"torch >= 2.0.1, != 2.3.0, < 2.5",
14+
"torch >= 2.0.1, != 2.3.0, != 2.5.0",
1515
"lightning >= 2.0",
1616
"s3torchconnector",
1717
"hydra-core",
18-
"torchdata>=0.6.1",
18+
"torchdata>=0.6.1, <=0.9.0", # we have dependency on deprecated DataPipes, which were removed in 0.10.0
1919
"torchvision",
2020
"s3fs>=2024.6.1",
2121
"transformers",

s3torchconnector/pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ classifiers = [
2222
]
2323

2424
dependencies = [
25-
"torch >= 2.0.1, < 2.5", # TODO: remove "< 2.5" restriction once https://github.com/pytorch/pytorch/issues/138333 is fixed
25+
"torch >= 2.0.1, != 2.5.0",
2626
"s3torchconnectorclient >= 1.3.0",
2727
]
2828

@@ -37,7 +37,7 @@ test = [
3737
]
3838

3939
e2e = [
40-
"torchdata",
40+
"torchdata<=0.9.0", # we have dependency on deprecated DataPipes, which were removed in 0.10.0
4141
"torchvision",
4242
"Pillow>=10.3.0",
4343
"boto3",
@@ -56,7 +56,7 @@ lightning-tests = [
5656

5757
dcp = [
5858
"tenacity",
59-
"torch >= 2.3, < 2.5", # TODO: remove "< 2.5" restriction once https://github.com/pytorch/pytorch/issues/138333 is fixed
59+
"torch >= 2.3, != 2.5.0",
6060
]
6161

6262
dcp-test = [

0 commit comments

Comments
 (0)