#!/bin/bash
#SBATCH --job-name=EHD_hw_loyo
#SBATCH --partition=workq
#SBATCH --nodes=1
#SBATCH --exclusive
#SBATCH --time=03:00:00
#SBATCH --output=/home/tanpv/Progs/claude/QG_1/logs/loyo_secondary_%j.out
#SBATCH --error=/home/tanpv/Progs/claude/QG_1/logs/loyo_secondary_%j.err
set -euo pipefail
ROOT=/home/tanpv/Progs/claude/QG_1
PY=/home/tanpv/.conda/envs/ehd/bin/python
export LD_LIBRARY_PATH=/home/tanpv/.conda/envs/ehd/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
cd "$ROOT"
mkdir -p logs outputs/loyo_pred_secondary

# Secondary target (heat-wave, >=3 consecutive EHDs). Sequential over the 20 LOYO folds in one
# process: obs + forecast years are preloaded once and reused across folds (memory-efficient).
export OMP_NUM_THREADS=${SLURM_CPUS_ON_NODE:-8} MKL_NUM_THREADS=${SLURM_CPUS_ON_NODE:-8} OPENBLAS_NUM_THREADS=${SLURM_CPUS_ON_NODE:-8}

echo "=== heat-wave LOYO (secondary target): 20 folds sequential @ $(hostname) $(date +%T) ==="
$PY scripts/06b_run_loyo_secondary.py
NP=$(ls outputs/loyo_pred_secondary/pred_hw_*.npz 2>/dev/null | wc -l)
echo "=== LOYO returned; pred files: $NP/20 ==="

echo "=== stage 08b verification $(date +%T) ==="
$PY scripts/08b_verify_secondary.py
echo "=== done $(date +%T) ==="