update env_setup

env_setup/create_env.sh

@@ -5,6 +5,12 @@ if [[ ! -n "$1" ]]; then
   exit 1
 fi
 
+if [[ ! -n "$2" ]]; then
+  echo "Provide the user name, which will be taken as folder name"
+  exit 1
+fi
+
+env_name=$2
 FOLDER_NAME=$1
 WORKING_DIR=/p/project/deepacf/deeprain/${FOLDER_NAME}/video_prediction_savp
 ENV_DIR=${WORKING_DIR}/env_setup
@@ -22,5 +28,6 @@ export PYTHONPATH=${ENV_DIR}/hickle/lib/python3.6/site-packages:$PYTHONPATH
 export PYTHONPATH=${WORKING_DIR}:$PYTHONPATH
 export PYTHONPATH=/p/home/jusers/${USER}/juwels/.local/bin:$PYTHONPATH
 export PYTHONPATH=${WORKING_DIR}/lpips-tensorflow:$PYTHONPATH
+jutil env activate -p $2