#!/bin/bash
# Waits for the existing X→Y→Z pipeline to finish, then runs W (Engelken L2).
set -e
cd /home/yurenh2/rrm/research/flossing
source /home/yurenh2/miniconda3/etc/profile.d/conda.sh
conda activate rrm
export CUDA_VISIBLE_DEVICES=2

PIPELINE_PID=$1
echo "[$(date '+%H:%M:%S')] W-runner waiting for pipeline PID $PIPELINE_PID..." >> step4_runner.log
while kill -0 "$PIPELINE_PID" 2>/dev/null; do
  sleep 30
done
echo "[$(date '+%H:%M:%S')] X→Y→Z done. Starting Phase W: Engelken L2 (k=4, α=1)" >> step4_runner.log

# Phase W: Engelken L2 — push ALL top-4 λ_i toward 0
python step4_from_scratch.py \
  --n-steps 3000 --batch-size 8 \
  --alpha-rf 1.0 --rf-mode engelken_l2 \
  --k-lyap 4 --lyap-act-steps 4 \
  --warmup-steps 200 \
  --eval-every 300 --eval-n 512 --eval-batch-size 32 \
  --out step4_W_engelken_l2_fromscratch.json \
  --save-ckpt ckpts/step4_W_final.pt \
  > step4_W.log 2>&1

echo "[$(date '+%H:%M:%S')] Phase W (Engelken L2) complete" >> step4_runner.log