#!/bin/bash
#SBATCH --job-name=EHD_WK
#SBATCH --nodes=1
#SBATCH --exclusive
#SBATCH --exclude=compute-1-2
#SBATCH --time=12:00:00
#SBATCH --array=0-11%10
#SBATCH --output=/home/tanpv/Progs/claude/HW/logs/wk_%A_%a.out
#SBATCH --error=/home/tanpv/Progs/claude/HW/logs/wk_%A_%a.err
set -euo pipefail
ROOT=/home/tanpv/Progs/claude/HW
PY=/home/tanpv/.conda/envs/ehd/bin/python
export LD_LIBRARY_PATH=/home/tanpv/.conda/envs/ehd/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
export OMP_NUM_THREADS=1 MKL_NUM_THREADS=1 PYTHONWARNINGS=ignore
cd "$ROOT"
# Tung tuan rieng W1..W6, hai dinh nghia su kien, mo hinh gop toan quoc, 541 o luoi
WIN=(W1 W2 W3 W4 W5 W6 W1 W2 W3 W4 W5 W6)
TGT=(so1 so1 so1 so1 so1 so1 geq_k geq_k geq_k geq_k geq_k geq_k)
i=$SLURM_ARRAY_TASK_ID
echo "=== task $i : ${WIN[$i]} / target ${TGT[$i]} / 541 o luoi / $(hostname) ==="
$PY src/phase1_arch.py --variant national --window "${WIN[$i]}" \
    --target "${TGT[$i]}" --eval all --jobs 20
echo "=== done ==="