From daef8d1680cce68a946b4f86e143081f857098d3 Mon Sep 17 00:00:00 2001
From: Bing Gong <b.gong@fz-juelich.de>
Date: Tue, 22 Feb 2022 11:52:32 +0100
Subject: [PATCH] revise create_env_non_HPC.sh

---
 video_prediction_tools/env_setup/create_env_non_HPC.sh | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/video_prediction_tools/env_setup/create_env_non_HPC.sh b/video_prediction_tools/env_setup/create_env_non_HPC.sh
index 708294fe..2adc5cf5 100644
--- a/video_prediction_tools/env_setup/create_env_non_HPC.sh
+++ b/video_prediction_tools/env_setup/create_env_non_HPC.sh
@@ -10,7 +10,7 @@ THIS_DIR="$(pwd)"
 WORKING_DIR="$(dirname "$THIS_DIR")"
 VENV_BASE=${WORKING_DIR}/virtual_envs
 VENV_DIR=${WORKING_DIR}/virtual_envs/${ENV_NAME}
-
+ACT_VENV="${VENV_DIR}/bin/activate"
 
 # check if directory to virtual environment is parsed
 if [ -z "$1" ]; then
@@ -40,6 +40,14 @@ export PYTHONPATH=${WORKING_DIR}/utils:$PYTHONPATH
 export PYTHONPATH=${WORKING_DIR}/model_modules:$PYTHONPATH
 export PYTHONPATH=${WORKING_DIR}/postprocess:$PYTHONPATH
 
+
+#ensure the PYTHONPATH is appended when activating the virtual enviornemnt
+echo 'export PYTHONPATH='${WORKING_DIR}':$PYTHONPATH' >> ${ACT_VENV}
+echo 'export PYTHONPATH='${WORKING_DIR}'/utils:$PYTHONPATH' >> ${ACT_VENV}
+echo 'export PYTHONPATH='${WORKING_DIR}'/model_modules:$PYTHONPATH' >> ${ACT_VENV}
+echo 'export PYTHONPATH='${WORKING_DIR}'/postprocess:$PYTHONPATH' >> ${ACT_VENV}
+
+
 # get back to basic directory
 cd "${WORKING_DIR}" || exit
 
-- 
GitLab