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

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

echo "[runpod-prepare] syncing dependencies with uv"
uv sync

echo "[runpod-prepare] ensuring transformers pinned version"
uv pip install transformers==4.53.2

VENV_SITE_PACKAGES="$(uv run python - <<'PY'
import site
paths = site.getsitepackages()
print(paths[0])
PY
)"

echo "[runpod-prepare] patching transformers with local replacements"
cp -r "${ROOT_DIR}/src/openpi/models_pytorch/transformers_replace/"* \
  "${VENV_SITE_PACKAGES}/transformers/"

echo "[runpod-prepare] done"