Fix sagemaker/pydantic incompatiblity causing test failures (#982)

pytorch · Dec 11, 2024 · edc0531 · edc0531
1 parent c8a633a
commit edc0531
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -22,7 +22,7 @@ pytest
 pytest-cov
 pytorch-lightning==2.3.1
 tensorboard==2.14.0
-sagemaker>=2.149.0
+sagemaker==2.224.4
 torch-model-archiver>=0.4.2
 torch==2.2.1
 torchmetrics==0.10.3