rebuild

Author: Jyothirmaikottu

data/common-safety-ignorelist.json

@@ -1,5 +1,6 @@
 {
     "51358": "Safety is not packaged with container, it is the result of security scanning process",
+    "85151": "Protobuf CVE in TensorFlow 2.19 - upstream dependency, cannot be patched without TF upgrade",
     "76769" : "A vulnerability, which was classified as problematic, was found in PyTorch 2.6.0. Which was built with container and cannot be patched.",
     "76771" : "A vulnerability, which was classified as problematic, was found in Pytorch version 2.5.1 and prior. Which was built with container and cannot be patched.",
     "78828" : "A DoS vulnerability in MKLDNN pooling implementation affecting PyTorch versions < 2.7.1-rc1. Container is PyTorch 2.6.0 and cannot be upgraded to 2.7.1."

tensorflow/training/docker/2.19/py3/Dockerfile.cpu

@@ -285,7 +285,7 @@ RUN uv pip install --system --no-cache-dir \
  && rm -rf /root/.cache/uv
 
 RUN uv pip install --system --no-cache-dir \
-    "sagemaker>=2,<3" \
+    "sagemaker>=2.256.0,<3" \
     sagemaker-experiments==0.1.45 \
     sagemaker-tensorflow-training \
     sagemaker-training \

tensorflow/training/docker/2.19/py3/Dockerfile.sagemaker.cpu.core_packages.json

@@ -13,7 +13,7 @@
     "version_specifier": ">=6.0,<6.1"
   },
   "sagemaker": {
-    "version_specifier": ">=2,<3"
+    "version_specifier": ">=2.256.0,<3"
   },
   "sagemaker-experiments": {
     "version_specifier": "<1"

tensorflow/training/docker/2.19/py3/cu125/Dockerfile.gpu

@@ -360,7 +360,7 @@ RUN uv pip install --system --no-cache-dir \
  && rm -rf /root/.cache/uv
 
 RUN uv pip install --system --no-cache-dir \
-    "sagemaker>=2,<3" \
+    "sagemaker>=2.256.0,<3" \
     sagemaker-experiments==0.1.45 \
     sagemaker-tensorflow-training \
     sagemaker-training \

tensorflow/training/docker/2.19/py3/cu125/Dockerfile.sagemaker.gpu.core_packages.json

@@ -12,7 +12,7 @@
     "version_specifier": ">=6.0,<6.1"
   },
   "sagemaker": {
-    "version_specifier": ">=2,<3"
+    "version_specifier": ">=2.256.0,<3"
   },
   "sagemaker-experiments": {
     "version_specifier": "<1"

test/requirements.txt

@@ -12,7 +12,7 @@ pytest-rerunfailures<=15.1
 pytest-timeout
 pytest-json-report
 pytest-xdist
-sagemaker>=2,<3
+sagemaker>=2.256.0,<3
 xmltodict
 retrying
 gitpython