add sniffio explicitly & rerun

Author: DevakiBolleneni

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

@@ -285,7 +285,9 @@ RUN pip install --no-cache-dir -U \
     "sagemaker==2.254.1" \
     sagemaker-experiments \
     sagemaker-pytorch-training \
-    sagemaker-training 
+    sagemaker-training \
+    # Add sniffio explicitly as it's not included in sagemaker==2.254.1 dependencies
+    sniffio  
 
 # Install extra packages
 RUN pip install --no-cache-dir -U \

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

@@ -253,7 +253,9 @@ RUN pip install --no-cache-dir -U \
     "sagemaker==2.254.1" \
     sagemaker-experiments \
     sagemaker-pytorch-training \
-    sagemaker-training 
+    sagemaker-training \
+    # Add sniffio explicitly as it's not included in sagemaker==2.254.1 dependencies
+    sniffio   
 
 # Install extra packages
 RUN pip install --no-cache-dir -U \