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

REPO_ID="${REPO_ID:-novastar111/pusht_336}"
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
HF_BIN="${HF_BIN:-/home/raychai/miniconda3/envs/llava_new/bin/hf}"

"${HF_BIN}" repo create "${REPO_ID}" --repo-type dataset --exist-ok
"${HF_BIN}" upload "${REPO_ID}" "${ROOT_DIR}" . \
  --repo-type dataset \
  --exclude "upload_to_hf.sh" \
  --exclude "logs/**" \
  --exclude "*_progress.json" \
  --exclude "*_worker_config.json" \
  --commit-message "Upload 336px PushT trajectory dataset"

echo "Uploaded dataset to https://huggingface.co/datasets/${REPO_ID}"