#!/bin/bash
# Push a checkpoint to R2
# Usage: push_to_r2.sh <checkpoint_dir> <step_number>
#
# Pushes to: r2:ptcheckpoints/cohere-transcribe-mixed/04-03-2026/ckpt-{step}/

CKPT_DIR="$1"
STEP="$2"
R2_PATH="r2:ptcheckpoints/cohere-transcribe-mixed/04-03-2026/ckpt-${STEP}/"

if [ -z "$CKPT_DIR" ] || [ -z "$STEP" ]; then
    echo "Usage: $0 <checkpoint_dir> <step_number>"
    exit 1
fi

echo "[R2] Pushing $CKPT_DIR → $R2_PATH"
rclone copy "$CKPT_DIR" "$R2_PATH" --progress --transfers 8
echo "[R2] Done: $R2_PATH"