[incubator-tvm] branch master updated: [DOCKER] Only pass pythonpath for ci images (#6005)

[tq...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git


The following commit(s) were added to refs/heads/master by this push:
     new 9f78c80  [DOCKER] Only pass pythonpath for ci images (#6005)
9f78c80 is described below

commit 9f78c805fc6bcab5bd2f122260994cd9e1102170
Author: Tianqi Chen <tq...@users.noreply.github.com>
AuthorDate: Tue Jul 7 12:53:51 2020 -0700

    [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[@]}