Skip to content

Commit

Permalink
[torchx/dev-requirements] Fix broken unittest in CI caused by new rel…
Browse files Browse the repository at this point in the history
…eases of sagemaker (2.235.2) and mlflow-skinny (2.18)
  • Loading branch information
kiukchung committed Nov 26, 2024
1 parent 26cb186 commit 874322a
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions dev-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -14,17 +14,19 @@ google-cloud-runtimeconfig==0.34.0
hydra-core
ipython
kfp==1.8.22
mlflow-skinny
mlflow-skinny<=2.17.2
moto~=5.0.8
pyre-extensions
pyre-check
pytest
pytest-cov
pytorch-lightning==2.3.1
tensorboard==2.14.0
sagemaker>=2.149.0
# sagemaker>=2.235.2 seems to be a bad release. APIs fail with
# NameError: Field name "json" shadows a BaseModel attribute; use a different field name with "alias='json'"
sagemaker>=2.149.0,<=2.235.1
torch-model-archiver>=0.4.2
torch==2.2.1
torch>=2.2.1
torchmetrics==0.10.3
torchserve>=0.10.0
torchtext==0.17.1
Expand All @@ -42,5 +44,4 @@ lintrunner-adapters
grpcio==1.62.1
grpcio-status==1.48.1
googleapis-common-protos==1.63.0
google-api-core==2.18.0
protobuf==3.20.3 # kfp==1.8.22 needs protobuf < 4

0 comments on commit 874322a

Please sign in to comment.