setup_improve.sh

#!/bin/bash --login
# Navigate to the dir with the cloned model repo
# Run it like this: source ./setup_improve.sh

# set -e

# Get current dir and model dir
model_path=$PWD
echo "Model path: $model_path"
model_name=$(echo "$model_path" | awk -F '/' '{print $NF}')
echo "Model name: $model_name"

# Download data (if needed)
data_dir="csa_data"
if [ ! -d $PWD/$data_dir/ ]; then
    echo "Download CSA data"
    source download_csa.sh
else
    echo "CSA data folder already exists"
fi

# Download author data (if needed) - PathDSP specific
author_dir="author_data"
if [ ! -d $PWD/$author_dir/ ]; then
    echo "Download author data"
    mkdir author_data
    source download_author_data.sh author_data/
else
    echo "Author data folder already exists"
fi

# Env var IMPROVE_DATA_DIR
export IMPROVE_DATA_DIR="./$data_dir/"

# Env var AUTHOR_DATA_DIR - PathDSP specific
export AUTHOR_DATA_DIR="./$author_dir/"

# Clone IMPROVE lib (if needed) and checkout the branch/tag
cd ../
improve_lib_path=$PWD/IMPROVE
# improve_branch="develop"
improve_branch="v0.1.0-2024-09-27"
if [ -d $improve_lib_path ]; then
    echo "IMPROVE repo exists in ${improve_lib_path}"
else
    git clone https://github.com/JDACS4C-IMPROVE/IMPROVE
fi
cd IMPROVE
git checkout -f $improve_branch 
cd ../$model_name

# Env var PYTHOPATH
export PYTHONPATH=$PYTHONPATH:$improve_lib_path

echo
echo "IMPROVE_DATA_DIR: $IMPROVE_DATA_DIR"
echo "AUTHOR_DATA_DIR: $AUTHOR_DATA_DIR"
echo "PYTHONPATH: $PYTHONPATH"