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

Author: jesszzzz

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