Terminate container when bootloader script fails (#1639)

Author: vishalbollu

images/onnx-predictor-cpu/Dockerfile

@@ -17,6 +17,8 @@ RUN cd /tmp/ && \
     cd - && \
     chmod +x /tmp/s6-overlay-amd64-installer && /tmp/s6-overlay-amd64-installer / && rm /tmp/s6-overlay-amd64-installer
 
+ENV S6_BEHAVIOUR_IF_STAGE2_FAILS 2
+
 RUN locale-gen en_US.UTF-8
 ENV LANG=en_US.UTF-8 LANGUAGE=en_US.en LC_ALL=en_US.UTF-8
 

images/onnx-predictor-gpu/Dockerfile

@@ -17,6 +17,8 @@ RUN cd /tmp/ && \
     cd - && \
     chmod +x /tmp/s6-overlay-amd64-installer && /tmp/s6-overlay-amd64-installer / && rm /tmp/s6-overlay-amd64-installer
 
+ENV S6_BEHAVIOUR_IF_STAGE2_FAILS 2
+
 RUN locale-gen en_US.UTF-8
 ENV LANG=en_US.UTF-8 LANGUAGE=en_US.en LC_ALL=en_US.UTF-8
 

images/python-predictor-cpu/Dockerfile

@@ -17,6 +17,8 @@ RUN cd /tmp/ && \
     cd - && \
     chmod +x /tmp/s6-overlay-amd64-installer && /tmp/s6-overlay-amd64-installer / && rm /tmp/s6-overlay-amd64-installer
 
+ENV S6_BEHAVIOUR_IF_STAGE2_FAILS 2
+
 RUN locale-gen en_US.UTF-8
 ENV LANG=en_US.UTF-8 LANGUAGE=en_US.en LC_ALL=en_US.UTF-8
 

images/python-predictor-gpu/Dockerfile

@@ -19,6 +19,8 @@ RUN cd /tmp/ && \
     cd - && \
     chmod +x /tmp/s6-overlay-amd64-installer && /tmp/s6-overlay-amd64-installer / && rm /tmp/s6-overlay-amd64-installer
 
+ENV S6_BEHAVIOUR_IF_STAGE2_FAILS 2
+
 RUN locale-gen en_US.UTF-8
 ENV LANG=en_US.UTF-8 LANGUAGE=en_US.en LC_ALL=en_US.UTF-8
 

images/python-predictor-inf/Dockerfile

@@ -13,6 +13,8 @@ RUN apt-get update -qq && apt-get install -y -q \
 RUN wget -P /tmp/ https://github.com/just-containers/s6-overlay/releases/download/v2.1.0.2/s6-overlay-amd64-installer && \
     chmod +x /tmp/s6-overlay-amd64-installer && /tmp/s6-overlay-amd64-installer / && rm /tmp/s6-overlay-amd64-installer
 
+ENV S6_BEHAVIOUR_IF_STAGE2_FAILS 2
+
 ENV PATH=/opt/aws/neuron/bin/:$PATH
 
 RUN apt-get update -qq && apt-get install -y -q \

images/tensorflow-predictor/Dockerfile

@@ -17,6 +17,8 @@ RUN cd /tmp/ && \
     cd - && \
     chmod +x /tmp/s6-overlay-amd64-installer && /tmp/s6-overlay-amd64-installer / && rm /tmp/s6-overlay-amd64-installer
 
+ENV S6_BEHAVIOUR_IF_STAGE2_FAILS 2
+
 RUN locale-gen en_US.UTF-8
 ENV LANG=en_US.UTF-8 LANGUAGE=en_US.en LC_ALL=en_US.UTF-8
 

pkg/workloads/cortex/serve/init/bootloader.sh

@@ -56,10 +56,6 @@ fi
 if [ -f "/mnt/project/dependencies.sh" ]; then
     eval $source_env_file_cmd
     bash -e /mnt/project/dependencies.sh
-    status=$?
-    if [ $status -ne 0 ]; then
-        exit $status
-    fi
 fi
 
 # install from conda-packages.txt