#!/usr/bin/env bash

while [[ "$#" -gt 0 ]]; do
    case "$1" in
        --data_dir)
            data_dir="$2"
            shift
            shift
            ;;
        *)
            echo "unknown parameter: $1"
            exit 1
            ;;
    esac
done

SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )

cat > .env <<EOL
export PROJECT_ROOT="${SCRIPT_DIR}"
export HYDRA_JOBS="${SCRIPT_DIR}"
export WANDB_DIR="${SCRIPT_DIR}"
export DATA_DIR="${data_dir}"
EOL

cp .env "${SCRIPT_DIR}/remote/DiffCSP-official"
cp .env "${SCRIPT_DIR}/remote/cdvae"