#!/usr/bin/env bash
#SBATCH --job-name=matrix-game2-minecraft-smoke
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=14
#SBATCH --gres=gpu:1
#SBATCH --time=04:00:00
#SBATCH --account=berzelius-2026-196
#SBATCH --chdir=/proj/cvl/users/x_fahkh2/worldmem-baseline-evals
#SBATCH --output=slurm_logs/%x-%A_%a.out
#SBATCH --error=slurm_logs/%x-%A_%a.out

set -euo pipefail

module load buildenv-gcccuda/12.9.1-gcc11

ROOT=/proj/cvl/users/x_fahkh2/worldmem-baseline-evals
ENV_ROOT=/proj/cvl/users/x_fahkh2/envs
REPO=$ROOT/Matrix-Game/Matrix-Game-2
PYTHON=$ENV_ROOT/matrix-game2-eval/bin/python
METRIC_PYTHON=$ENV_ROOT/geometry-forcing-eval/bin/python
WORLD_MEM_ROOT=$ROOT/references/worldmem
DATA_ROOT=/proj/cvl/users/x_fahkh2/WorldMem_Repro/datasets/minecraft
MANIFEST=$ROOT/deployment/manifests/minecraft_test_300.csv
SEED=42
FRAMES=24
RUN=matrix-game2_minecraft_smoke_job${SLURM_JOB_ID}
OUTPUT=$REPO/outputs/$RUN

export TMPDIR=$ROOT/cache
export HF_HOME=$ROOT/cache/huggingface
export TORCH_HOME=$ROOT/cache/torch
export XDG_CACHE_HOME=$ROOT/cache
export MPLCONFIGDIR=$ROOT/cache/matplotlib
export TRITON_CACHE_DIR=$ROOT/cache
export PYTHONNOUSERSITE=1
export PYTHONUNBUFFERED=1
export OMP_NUM_THREADS=4
export MKL_NUM_THREADS=4
export WANDB_MODE=disabled
export HF_HUB_OFFLINE=1
export TRANSFORMERS_OFFLINE=1
mkdir -p "$TMPDIR" "$OUTPUT"
cd "$REPO"

"$PYTHON" "$ROOT/deployment/verify_node.py" --expected-gpus 1 --output "$OUTPUT"
cp "$0" "$OUTPUT/launcher.sh"

srun --ntasks=1 --gpus-per-task=1 --cpus-per-task=14 --cpu-bind=cores \
    "$PYTHON" "$ROOT/shared/run_and_score.py" \
    --method matrix-game2 \
    --workspace-root "$ROOT" \
    --metric-python "$METRIC_PYTHON" \
    --worldmem-root "$WORLD_MEM_ROOT" \
    --manifest "$MANIFEST" \
    --data-root "$DATA_ROOT" \
    --output "$OUTPUT" \
    --scope monitor8 \
    --seed "$SEED" \
    --limit 1 \
    --max-generated "$FRAMES" \
    --rank 0 \
    --world-size 1 \
    --keep-tensor

srun --exclusive --exact --nodes=1 --ntasks=1 --gpus-per-task=1 --cpus-per-task=14 \
    "$METRIC_PYTHON" "$ROOT/shared/aggregate.py" \
    --output "$OUTPUT" \
    --manifest "$MANIFEST" \
    --dataset minecraft \
    --scope monitor8 \
    --seed "$SEED" \
    --expected-frames "$FRAMES" \
    --limit 1

printf 'Smoke evaluation completed: %s\n' "$OUTPUT"