try prod sm pins

Author: Jyothirmaikottu

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

@@ -132,7 +132,7 @@ RUN wget https://www.python.org/ftp/python/$PYTHON_VERSION/Python-$PYTHON_VERSIO
 
 RUN ${PIP} --no-cache-dir install --upgrade \
     pip \
-    "setuptools>=81.0.0"
+    setuptools
 
 # Some TF tools expect a "python" binary
 RUN ln -s $(which ${PYTHON}) /usr/local/bin/python \
@@ -289,27 +289,22 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
     shap
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
-    sagemaker \
-    sagemaker-experiments \
-    sagemaker-tensorflow-training \
-    sagemaker-training \
-    y-py \
-    sagemaker-studio-analytics-extension \
-    "sparkmagic<1" \
-    sagemaker-studio-sparkmagic-lib \
-    smclarify
+    sagemaker==2.254.1 \
+    sagemaker-experiments==0.1.45 \
+    sagemaker-tensorflow-training==20.4.1 \
+    sagemaker-training==5.1.1 \
+    y-py==0.6.2 \
+    sagemaker-studio-analytics-extension==0.1.7 \
+    sparkmagic==0.22.0 \
+    sagemaker-studio-sparkmagic-lib==0.1.4 \
+    smclarify==0.5
 
 # Remove python kernel installed by sparkmagic
 RUN /usr/local/bin/jupyter-kernelspec remove -f python3
 
 # Remove ipykernel
 RUN rm -rf /usr/local/share/jupyter/kernels/python3/kernel.json
 
-# Clean up JupyterLab staging and test directories to remove CVEs
-RUN rm -rf /usr/local/lib/python*/site-packages/jupyterlab/staging \
-    && rm -rf /usr/local/lib/python*/site-packages/jupyterlab/tests \
-    && rm -rf /usr/local/share/jupyter/lab/staging
-
 # remove tmp files
 RUN rm -rf /tmp/*
 

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

@@ -193,7 +193,7 @@ RUN wget https://www.python.org/ftp/python/$PYTHON_VERSION/Python-$PYTHON_VERSIO
 
 RUN ${PIP} --no-cache-dir install --upgrade \
    pip \
-   "setuptools>=81.0.0" \
+   setuptools \
    wheel
 
 # Some TF tools expect a "python" binary
@@ -392,11 +392,6 @@ RUN /usr/local/bin/jupyter-kernelspec remove -f python3
 # Remove ipykernel
 RUN rm -rf /usr/local/share/jupyter/kernels/python3/kernel.json
 
-# Clean up JupyterLab staging and test directories to remove CVEs
-RUN rm -rf /usr/local/lib/python*/site-packages/jupyterlab/staging \
-    && rm -rf /usr/local/lib/python*/site-packages/jupyterlab/tests \
-    && rm -rf /usr/local/share/jupyter/lab/staging
-
 # remove tmp files
 RUN rm -rf /tmp/*