#!/usr/bin/env bash
# Clone the base codebases Flow-CoPD depends on (PINNED commits for reproducibility).
# Run this once on the training machine after unpacking the repo.
set -euo pipefail
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
mkdir -p "$ROOT/third_party"; cd "$ROOT/third_party"

FLOW_GRPO_COMMIT=879042cf5707f8b90daa98d147d7deac2317c5da   # 2026-05-07
NFT_COMMIT=cbb14f84b8312b620390dfcbe2fab69c1383b104

if [ ! -d flow_grpo/.git ]; then
  echo ">>> cloning flow_grpo (base codebase + prompt datasets)"
  git clone https://github.com/yifan123/flow_grpo.git
  git -C flow_grpo checkout "$FLOW_GRPO_COMMIT"
fi
if [ ! -d DiffusionNFT/.git ]; then
  echo ">>> cloning DiffusionNFT (Block B2 baseline)"
  git clone https://github.com/NVlabs/DiffusionNFT.git
  git -C DiffusionNFT checkout "$NFT_COMMIT" || echo "(NFT commit checkout skipped; using default branch)"
fi
echo "DONE. third_party/flow_grpo@${FLOW_GRPO_COMMIT}  third_party/DiffusionNFT@${NFT_COMMIT}"
echo "Prompt datasets are now at third_party/flow_grpo/dataset/{pickscore,ocr,geneval}/"