#!/bin/bash
#SBATCH --account=<your-slurm-account>
#SBATCH --partition=l40s
#SBATCH --nodes=1
#SBATCH --ntasks=8
#SBATCH --gres=gpu:l40s:1
#SBATCH --time=24:00:00
#SBATCH --job-name=ddpm_poster
#SBATCH --mail-user=<your-email>   # replace before submitting
#SBATCH --output=slurm-ddpm-poster-%j.out
#SBATCH --error=slurm-ddpm-poster-%j.err

# Corrected six-anchor surrogate posteriors (poster.py): DDPM-2 + DDPM-6 with
# stochastic averaging, calibrated sigma_pk, MC marginalisation for 6-param, etc.
#
# Submit from anywhere:
#   sbatch <DDPM_ROOT>/Models/scripts/run_poster.sh
#
# Override output directory:
#   sbatch --export=OUTPUT_DIR=<DDPM_ROOT>/Models/my_poster_out \
#     <DDPM_ROOT>/Models/scripts/run_poster.sh
#
# Extra CLI passes through to poster.py, e.g. DDPM-2 only (faster debug):
#   sbatch .../run_poster.sh --ddpm2-only --grid 14 --n-pk-samples 4 --n-marg-samples 1 --no-ppc
#
# Interactive (same module as other Models scripts):
#   module load python/miniconda3-py3.12-usr
#   bash <DDPM_ROOT>/Models/scripts/run_poster.sh --help

set -euo pipefail

ROOT="<DDPM_ROOT>/Models"
cd "$ROOT"

module load python/miniconda3-py3.12-usr

OUT="${OUTPUT_DIR:-${ROOT}/ddpm_posterior_corrected_out}"

echo "==============================================="
echo "Job ID: ${SLURM_JOB_ID:-local}"
echo "Job Name: ${SLURM_JOB_NAME:-run_poster}"
echo "Node: ${SLURM_NODELIST:-$(hostname)}"
echo "GPU: ${CUDA_VISIBLE_DEVICES:-n/a}"
echo "Starting Time: $(date)"
echo "Poster output: ${OUT}"
echo "==============================================="

python "$ROOT/poster.py" --output-dir "$OUT" "$@"

echo "==============================================="
echo "Artifacts -> ${OUT}"
echo "Finished at: $(date)"
echo "==============================================="