#!/usr/bin/env bash
set -euo pipefail

GPU="${1:-0}"
ROOT="/home/yurenh2/rrm"
PY="/home/yurenh2/miniconda3/envs/rrm/bin/python"
OUT_DIR="${ROOT}/research/flossing/flossing_suite/results/smoke"
mkdir -p "${OUT_DIR}"
cd "${ROOT}"

echo "[smoke] toy RNN faithful port"
CUDA_VISIBLE_DEVICES="${GPU}" PYTHONUNBUFFERED=1 "${PY}" research/flossing/engelken_python_flossing.py \
  --hidden-size 16 \
  --n-lyap 4 \
  --train-epochs 3 \
  --pre-epochs 2 \
  --inter-epochs 0 \
  --max-inter-episodes 0 \
  --batch-size 4 \
  --train-steps 40 \
  --lyap-steps 30 \
  --floss-input-steps 40 \
  --eval-every 1 \
  --eval-batches 1 \
  --device cuda \
  --out "${OUT_DIR}/toy_smoke.json" \
  > "${OUT_DIR}/toy_smoke.log" 2>&1

echo "[smoke] TRM interfloss analogue"
CUDA_VISIBLE_DEVICES="${GPU}" PYTHONUNBUFFERED=1 "${PY}" research/flossing/step7_interfloss.py \
  --model trm \
  --ckpt-root "${ROOT}/trm/checkpoints/Sudoku-extreme-1k-aug-1000-ACT-torch/pretrain_mlp_t_sudoku_official_gbs768_repro" \
  --ckpt-name __random__ \
  --init-seed 123 \
  --train-steps 1 \
  --batch-size 2 \
  --train-lr 1e-4 \
  --floss-lr 1e-4 \
  --floss-steps 1 \
  --interfloss-at 0 \
  --floss-mode engelken_l2 \
  --lambda-star 0 \
  --k-lyap 1 \
  --lyap-act-steps 1 \
  --seed 42 \
  --eval-every 1 \
  --eval-n 8 \
  --eval-batch-size 8 \
  --floss-log-every 1 \
  --no-eval-after-floss \
  --train-puzzle-emb \
  --puzzle-emb-lr 1e-4 \
  --puzzle-emb-weight-decay 1.0 \
  --out "${OUT_DIR}/trm_step7_smoke.json" \
  > "${OUT_DIR}/trm_step7_smoke.log" 2>&1

echo "[smoke] summarizer"
"${PY}" research/flossing/flossing_suite/summarize_flossing.py \
  > "${OUT_DIR}/summarize_smoke.log" 2>&1

echo "smoke OK: ${OUT_DIR}"