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

jesszzzz · web-flow · commit edc05316a38f · 2024-12-11T14:33:03.000-05:00
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
