From 9f78c805fc6bcab5bd2f122260994cd9e1102170 Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Tue, 7 Jul 2020 12:53:51 -0700
Subject: [PATCH] [DOCKER] Only pass pythonpath for ci images (#6005)

---
 docker/bash.sh | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/docker/bash.sh b/docker/bash.sh
index a88951f..191448f 100755
--- a/docker/bash.sh
+++ b/docker/bash.sh
@@ -69,6 +69,12 @@ else
     DOCKER_BINARY="docker"
 fi
 
+if [[ "${DOCKER_IMAGE_NAME}" == *"ci"* ]]; then
+    CI_PY_ENV="-e PYTHONPATH=/workspace/python:/workspace/topi/python"
+else
+    CI_PY_ENV=""
+fi
+
 # Print arguments.
 echo "WORKSPACE: ${WORKSPACE}"
 echo "DOCKER CONTAINER NAME: ${DOCKER_IMAGE_NAME}"
@@ -98,10 +104,10 @@ ${DOCKER_BINARY} run --rm --pid=host\
     -e "CI_BUILD_UID=$(id -u)" \
     -e "CI_BUILD_GROUP=$(id -g -n)" \
     -e "CI_BUILD_GID=$(id -g)" \
-    -e "PYTHONPATH=/workspace/python:/workspace/topi/python"\
     -e "CI_PYTEST_ADD_OPTIONS=$CI_PYTEST_ADD_OPTIONS" \
-    ${CUDA_ENV}\
+    ${CI_PY_ENV} \
+    ${CUDA_ENV} \
     ${CI_DOCKER_EXTRA_PARAMS[@]} \
-    ${DOCKER_IMAGE_NAME}\
+    ${DOCKER_IMAGE_NAME} \
     bash --login /docker/with_the_same_user \
     ${COMMAND[@]}
-- 
2.7.4