From 807da53da3408bb2ed558cf781c79e655ef4b1b0 Mon Sep 17 00:00:00 2001
From: masak1112 <gongbing1112@gmail.com>
Date: Fri, 25 Feb 2022 11:34:42 +0100
Subject: [PATCH] fix the bugs when run train script and load moudles issues
---
video_prediction_tools/env_setup/requirements.txt | 2 +-
video_prediction_tools/env_setup/wrapper_container.sh | 5 ++++-
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/video_prediction_tools/env_setup/requirements.txt b/video_prediction_tools/env_setup/requirements.txt
index 35f6eb24..c7eec5e0 100755
--- a/video_prediction_tools/env_setup/requirements.txt
+++ b/video_prediction_tools/env_setup/requirements.txt
@@ -7,7 +7,7 @@ numpy==1.17.3 # although this numpy-version is in the container, we set it h
scikit-image==0.18.1
opencv-python-headless==4.2.0.34
netcdf4==1.5.8
-metadata==0.2
+#metadata==0.2
normalization==0.4
utils==1.0.1
diff --git a/video_prediction_tools/env_setup/wrapper_container.sh b/video_prediction_tools/env_setup/wrapper_container.sh
index cfe716be..97d39be8 100755
--- a/video_prediction_tools/env_setup/wrapper_container.sh
+++ b/video_prediction_tools/env_setup/wrapper_container.sh
@@ -8,7 +8,10 @@ VENV_DIR=$WORKING_DIR/virtual_envs/$1
shift # replaces $1 by $2, so that $@ does not include the name of the virtual environment anymore
# sanity checks
-if [[ "${EXE_DIR}" != "HPC_scripts" ]]; then
+if [[ "${EXE_DIR}" = "HPC_scripts" ]] || [[ "${EXE_DIR}" = "JSC_scripts" ]];
+then
+ echo "The runscript is running under the folder ${EXE_DIR}"
+else
echo "ERROR: Run the setup-script for the enviornment from the HPC_scripts-directory!"
exit
fi
--
GitLab