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

ROOT_DIR="${ROOT_DIR:-/workspace}"

echo "[setup] expected hardware: NVIDIA RTX PRO 6000 Blackwell Server Edition"
echo "[setup] expected OS family: Ubuntu 24.04 / Linux 6.8"

if [ ! -d "$ROOT_DIR/VLAarchtests" ]; then
  echo "[setup] expected staged tree at $ROOT_DIR/VLAarchtests" >&2
fi

echo "[setup] base python:"
python --version || true

if [ -x "$ROOT_DIR/.tools/micromamba/bin/micromamba" ]; then
  echo "[setup] micromamba envs:"
  "$ROOT_DIR/.tools/micromamba/bin/micromamba" env list || true
fi

echo "[setup] current GPU snapshot:"
nvidia-smi --query-gpu=name,driver_version,memory.total --format=csv,noheader || true

echo "[setup] this repo includes package snapshots in environment/"
echo "[setup] recreate the rlbench env at $ROOT_DIR/envs/rlbench before running the overlap baseline path"