#!/bin/bash
#SBATCH --job-name=DrSFC_AI_BC
#SBATCH --nodes=6
#SBATCH --ntasks-per-node=20
#SBATCH --cpus-per-task=1
#SBATCH --output=/home/tanpv/Progs/CLI/Dr/logs/ai_bias_correction_%j.out
#SBATCH --error=/home/tanpv/Progs/CLI/Dr/logs/ai_bias_correction_%j.err

set -euo pipefail
PROJECT_ROOT=/home/tanpv/Progs/CLI/Dr
PYTHON_BIN=/home/hapt/miniconda3/bin/python3
RUN_ID=${SLURM_JOB_ID}
cd "${PROJECT_ROOT}"
export OMP_NUM_THREADS=1
export MKL_NUM_THREADS=1
export OPENBLAS_NUM_THREADS=1

"${PYTHON_BIN}" ai_bias_correction.py preflight --run "${RUN_ID}"
srun --kill-on-bad-exit=1 "${PYTHON_BIN}" ai_bias_correction.py worker --run "${RUN_ID}"
"${PYTHON_BIN}" ai_bias_correction.py finalize --run "${RUN_ID}"