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

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "$ROOT_DIR"

export OPENCLAUDE_MOCK="${OPENCLAUDE_MOCK:-1}"

OUT_ROOT="${DOVLA_DEBUG_ROOT:-outputs/phase5_train_debug}"
TASKS_PATH="$OUT_ROOT/tasks.jsonl"
DATASET_DIR="$OUT_ROOT/cil"
RUN_DIR="$OUT_ROOT/run"

mkdir -p "$OUT_ROOT"

python scripts/generate_tasks.py --mock --num-tasks 3 --out "$TASKS_PATH" --seed 0
python scripts/generate_cil.py \
  --backend toy \
  --tasks "$TASKS_PATH" \
  --out "$DATASET_DIR" \
  --num-states-per-task 2 \
  --k 4 \
  --seed 0 \
  --shard-size 8 \
  --inline-observations
python scripts/train_dovla.py \
  --dataset "$DATASET_DIR" \
  --out "$RUN_DIR" \
  --epochs 1 \
  --batch-groups 2 \
  --records-per-group 4 \
  --hidden-dim 64 \
  --lr 0.001 \
  --device "${DOVLA_DEVICE:-auto}" \
  --seed 0

echo "debug training run: $RUN_DIR"