add /opt/ml/model/code to PYTHONPATH for SDK-packaged tarballs

Author: jinyan-li1

scripts/ray/dockerd_entrypoint.sh

@@ -12,8 +12,8 @@ else
 fi
 echo "Detected ${NUM_GPUS} GPU(s)"
 
-# Add model directory to Python path
-export PYTHONPATH="/opt/ml/model:${PYTHONPATH}"
+# Add model directory and SDK code/ subdir to Python path
+export PYTHONPATH="/opt/ml/model:/opt/ml/model/code:${PYTHONPATH}"
 
 # Install runtime requirements if present
 if [ -f "/opt/ml/model/code/requirements.txt" ]; then

scripts/ray/sagemaker_entrypoint.sh

@@ -19,8 +19,8 @@ echo "Detected ${NUM_GPUS} GPU(s)"
 if [ $# -eq 0 ] || [ "$1" = "serve" ]; then
     echo "Starting Ray Serve for SageMaker..."
 
-    # Add model directory to Python path before ray start
-    export PYTHONPATH="/opt/ml/model:${PYTHONPATH}"
+    # Add model directory and SDK code/ subdir to Python path before ray start
+    export PYTHONPATH="/opt/ml/model:/opt/ml/model/code:${PYTHONPATH}"
 
     # Install runtime requirements before Ray starts
     if [ -f "/opt/ml/model/code/requirements.txt" ]; then