update dockerfile

aws · Jan 27, 2025 · 6f87898 · 6f87898
1 parent 799ce65
commit 6f87898
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/build_artifacts/v2/v2.3/v2.3.0/Dockerfile b/build_artifacts/v2/v2.3/v2.3.0/Dockerfile
@@ -95,12 +95,15 @@ RUN if [[ -z $ARG_BASED_ENV_IN_FILENAME ]] ; \
     micromamba install -y --name base --file /tmp/$ENV_IN_FILENAME && \
     mkdir -p $SAGEMAKER_RECOVERY_MODE_HOME && \
     chown $MAMBA_USER:$MAMBA_USER $SAGEMAKER_RECOVERY_MODE_HOME && \
-    SUPERVISOR_VERSION=$(grep "^conda-forge::supervisor\[" /tmp/$ENV_IN_FILENAME) && \
-    JUPYTERLAB_VERSION=$(grep "^conda-forge::jupyterlab\[" /tmp/$ENV_IN_FILENAME) && \
-    SAGEMAKER_JUPYTERLAB_VERSION=$(grep "^conda-forge::sagemaker-jupyterlab-extension" /tmp/$ENV_IN_FILENAME) && \
-    echo "Installing in sagemaker-recovery-mode micromamba environment: $JUPYTERLAB_VERSION $SAGEMAKER_JUPYTERLAB_VERSION" && \
+    SUPERVISOR_VERSION=$(micromamba list | grep 'supervisor ' | tr -s ' ' | cut -d ' ' -f 3 | head -n 1) && \
+    JUPYTERLAB_VERSION=$(micromamba list | grep 'jupyterlab ' | tr -s ' ' | cut -d ' ' -f 3 | head -n 1) && \
+    SAGEMAKER_JUPYTERLAB_VERSION=$(micromamba list | grep 'sagemaker-jupyterlab-extension ' | tr -s ' ' | cut -d ' ' -f 3 | head -n 1) && \
+    echo "Installing in sagemaker-recovery-mode micromamba environment: jupyterlab $JUPYTERLAB_VERSION sagemaker-jupyterlab-extension $SAGEMAKER_JUPYTERLAB_VERSION" && \
     micromamba create -n sagemaker-recovery-mode && \
-    micromamba install -n sagemaker-recovery-mode -y $JUPYTERLAB_VERSION $SAGEMAKER_JUPYTERLAB_VERSION $SUPERVISOR_VERSION && \
+    micromamba install -n sagemaker-recovery-mode -y -c conda-forge \
+        jupyterlab==$JUPYTERLAB_VERSION \
+        sagemaker-jupyterlab-extension==$SAGEMAKER_JUPYTERLAB_VERSION \
+        supervisor==$SUPERVISOR_VERSION && \
     micromamba clean --all --yes --force-pkgs-dirs && \
     rm -rf /tmp/*.in && \
     sudo ln -s $(which python3) /usr/bin/python && \
@@ -182,7 +185,6 @@ RUN mkdir -p $SAGEMAKER_LOGGING_DIR && \
     chmod a+rw $SAGEMAKER_LOGGING_DIR && \
     mkdir -p ${STUDIO_LOGGING_DIR} && \
     chown ${NB_USER}:${MAMBA_USER} ${STUDIO_LOGGING_DIR} && \
-
     # Clean up CodeEditor artifacts
     rm -rf /etc/code-editor && \
     # Create supervisord runtime directory