update scripts for recovery mode

aws · Jan 13, 2025 · ae511d0 · ae511d0
1 parent 188ff02
commit ae511d0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/build_artifacts/v2/v2.3/v2.3.0/dirs/usr/local/bin/entrypoint-jupyter-server b/build_artifacts/v2/v2.3/v2.3.0/dirs/usr/local/bin/entrypoint-jupyter-server
@@ -6,8 +6,14 @@ set -e
 # micromamba commands (e.g. using `micromamba activate` to activate environments)
 eval "$(micromamba shell hook --shell=bash)"
 
-# Activate conda environment 'base', where supervisord is installed
-micromamba activate base
+if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then
+    export HOME=$SAGEMAKER_RECOVERY_MODE
+    # Activate conda environment `sagemaker-recovery-mode`
+    micromamba activate sagemaker-recovery-mode
+else
+    # Activate conda environment 'base'
+    micromamba activate base
+fi
 
 # Set up SAGEMAKER_APP_TYPE_LOWERCASE based on SAGEMAKER_APP_TYPE
 export SAGEMAKER_APP_TYPE_LOWERCASE=$(echo $SAGEMAKER_APP_TYPE | tr '[:upper:]' '[:lower:]')

diff --git a/build_artifacts/v2/v2.3/v2.3.0/dirs/usr/local/bin/start-jupyter-server b/build_artifacts/v2/v2.3/v2.3.0/dirs/usr/local/bin/start-jupyter-server
@@ -3,8 +3,14 @@ set -e
 
 eval "$(micromamba shell hook --shell=bash)"
 
-# Activate conda environment 'base', which is the default environment for Cosmos
-micromamba activate base
+if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then
+    export HOME=$SAGEMAKER_RECOVERY_MODE
+    # Activate conda environment `sagemaker-recovery-mode`
+    micromamba activate sagemaker-recovery-mode
+else
+    # Activate conda environment 'base'
+    micromamba activate base
+fi
 
 # Start Jupyter server in rtc mode for shared spaces
 if [ -n "$SAGEMAKER_APP_TYPE_LOWERCASE" ] && [ "$SAGEMAKER_SPACE_TYPE_LOWERCASE" == "shared" ]; then