pin sagemaker to <3

Author: sirutBuasai

pytorch/training/docker/2.8/py3/Dockerfile.cpu

@@ -287,7 +287,7 @@ WORKDIR /
 # Install SM packages
 RUN pip install --no-cache-dir -U \
     smclarify \
-    sagemaker \
+    "sagemaker<3" \
     sagemaker-experiments \
     sagemaker-pytorch-training \
     sagemaker-training \

pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu

@@ -255,7 +255,7 @@ RUN pip install --no-cache-dir -U \
     "awscli<1.42.50" \
     "boto3<1.40.50" \
     smclarify \
-    sagemaker \
+    "sagemaker<3" \
     sagemaker-experiments \
     sagemaker-pytorch-training \
     sagemaker-training \