Skip to content

Commit f60a2bd

Browse files
author
Kiuk Chung
committed
[torchx/dev-requirements] Fix broken unittest in CI caused by new releases of sagemaker (2.235.2)
1 parent 26cb186 commit f60a2bd

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

dev-requirements.txt

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,11 @@ pytest
2222
pytest-cov
2323
pytorch-lightning==2.3.1
2424
tensorboard==2.14.0
25-
sagemaker>=2.149.0
25+
# sagemaker>=2.235.2 seems to be a bad release. APIs fail with
26+
# NameError: Field name "json" shadows a BaseModel attribute; use a different field name with "alias='json'"
27+
sagemaker>=2.149.0,<=2.235.1
2628
torch-model-archiver>=0.4.2
27-
torch==2.2.1
29+
torch>=2.2.1
2830
torchmetrics==0.10.3
2931
torchserve>=0.10.0
3032
torchtext==0.17.1
@@ -42,5 +44,4 @@ lintrunner-adapters
4244
grpcio==1.62.1
4345
grpcio-status==1.48.1
4446
googleapis-common-protos==1.63.0
45-
google-api-core==2.18.0
4647
protobuf==3.20.3 # kfp==1.8.22 needs protobuf < 4

0 commit comments

Comments
 (0)