run_model_slurm.sh

#!/bin/bash

CONDA_BASE=$(conda info --base)
source $CONDA_BASE/etc/profile.d/conda.sh
while [[ -v CONDA_DEFAULT_ENV ]]; do
    conda deactivate
done

i=1
RUN_MODEL_OPTS=()
while [[ $i -le $# ]]; do
    if [[ ${!i} == "--n-jobs="* ]]; then
        N_JOBS=${!i#*=}
        RUN_MODEL_OPTS+=(${!i})
    elif [[ ${!i} == "--n-jobs" ]]; then
        RUN_MODEL_OPTS+=(${!i})
        i=$((i + 1))
        N_JOBS=${!i}
        RUN_MODEL_OPTS+=($N_JOBS)
    elif [[ ${!i} == "--sbatch-opts="* ]]; then
        SBATCH_OPTS=${!i#*=}
    elif [[ ${!i} == "--sbatch-opts" ]]; then
        i=$((i + 1))
        SBATCH_OPTS=${!i}
    else
        RUN_MODEL_OPTS+=(${!i})
    fi
    i=$((i + 1))
done

if [[ ! -v N_JOBS ]]; then
    # most common Biowulf node has 56 CPUs
    N_JOBS=56
    RUN_MODEL_OPTS+=("--n-jobs $N_JOBS")
fi

SCRIPT_DIR=$(dirname $(realpath -s $0))

SBATCH_CMD="sbatch \
--chdir=$(realpath $SCRIPT_DIR) \
--cpus-per-task=$N_JOBS \
$SBATCH_OPTS \
$SCRIPT_DIR/run_model.sh ${RUN_MODEL_OPTS[@]}"
echo $SBATCH_CMD
$SBATCH_CMD