unpin sagemaker

Author: sirutBuasai

pytorch/training/buildspec-2-8-sm.yml

@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
 version: &VERSION 2.8.0
 short_version: &SHORT_VERSION "2.8"
 arch_type: x86
-autopatch_build: "True"
+# autopatch_build: "True"
 
 repository_info:
   training_repository: &TRAINING_REPOSITORY

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==2.254.1" \
+    sagemaker \
     sagemaker-experiments \
     sagemaker-pytorch-training \
     sagemaker-training \

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

@@ -255,12 +255,12 @@ RUN pip install --no-cache-dir -U \
     "awscli<1.42.50" \
     "boto3<1.40.50" \
     smclarify \
-    "sagemaker==2.254.1" \
+    sagemaker \
     sagemaker-experiments \
     sagemaker-pytorch-training \
     sagemaker-training \
     # Add sniffio explicitly as it's not included in sagemaker==2.254.1 dependencies
-    sniffio   
+    sniffio
 
 # Install extra packages
 RUN pip install --no-cache-dir -U \