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

REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
source "${REPO_ROOT}/environment/runtime_env_vars.sh"

PYTHONPATH_R3D="/workspace/venv_r3d/lib/python3.11/site-packages:${REPO_ROOT}/code/reveal_vla_bimanual:/usr/local/lib/python3.11/dist-packages"

echo "[validate] pytest"
PYTHONPATH="${PYTHONPATH_R3D}" python -m pytest -q "${REPO_ROOT}/tests"

echo "[validate] rlbench smoke_test"
/workspace/.tools/micromamba/bin/micromamba run -r /workspace/.micromamba -p /workspace/envs/rlbench \
  bash -lc "cd /workspace/reveal_vla_bimanual && PYTHONPATH=/workspace/reveal_vla_bimanual python -m sim_rlbench.smoke_test --print-train-command"

echo "[validate] rlbench launch_smoke open_drawer"
/workspace/.tools/micromamba/bin/micromamba run -r /workspace/.micromamba -p /workspace/envs/rlbench \
  bash -lc "cd /workspace/reveal_vla_bimanual && PYTHONPATH=/workspace/reveal_vla_bimanual python -m sim_rlbench.launch_smoke --task open_drawer --headless"