test uv: sm < 3 pin

Author: Jyothirmaikottu

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 \
+    "sagemaker>=2,<3" \
     sagemaker-experiments==0.1.45 \
     sagemaker-tensorflow-training \
     sagemaker-training \

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

@@ -12,6 +12,12 @@
   "pyyaml": {
     "version_specifier": ">=6.0,<6.1"
   },
+  "sagemaker": {
+    "version_specifier": ">=2,<3"
+  },
+  "sagemaker-experiments": {
+    "version_specifier": "<1"
+  },
   "sparkmagic": {
     "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 \
+    "sagemaker>=2,<3" \
     sagemaker-experiments==0.1.45 \
     sagemaker-tensorflow-training \
     sagemaker-training \

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

@@ -11,8 +11,11 @@
   "pyyaml": {
     "version_specifier": ">=6.0,<6.1"
   },
+  "sagemaker": {
+    "version_specifier": ">=2,<3"
+  },
   "sagemaker-experiments": {
-    "version_specifier": "==0.*"
+    "version_specifier": "<1"
   },
   "sparkmagic": {
     "version_specifier": "<1"