#!/bin/bash
#SBATCH --job-name=dovla_ms_collect
#SBATCH --account=def-yalda_cpu
#SBATCH --partition=cpubase_bycore_b1
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=2
#SBATCH --mem=8G
#SBATCH --time=00:20:00
#SBATCH --output=outputs/hpc/logs/%x_%j.out
#SBATCH --error=outputs/hpc/logs/%x_%j.err

set -euo pipefail

PROJECT_DIR="${PROJECT_DIR:-$SLURM_SUBMIT_DIR}"
PICKCUBE_DATA="${PICKCUBE_DATA:?Set PICKCUBE_DATA}"
MULTITASK_ROOT="${MULTITASK_ROOT:?Set MULTITASK_ROOT}"
COLLECTION_OUT="${COLLECTION_OUT:?Set COLLECTION_OUT}"
COLLECTION_NAME="${COLLECTION_NAME:-maniskill-six-task-k16}"
PYTHON="${PYTHON:-$PROJECT_DIR/.venv/bin/python}"

cd "$PROJECT_DIR"
"$PYTHON" scripts/make_cil_collection.py \
  --name "$COLLECTION_NAME" \
  --out "$COLLECTION_OUT" \
  --sources \
    "$PICKCUBE_DATA" \
    "$MULTITASK_ROOT/PushCube-v1" \
    "$MULTITASK_ROOT/PullCube-v1" \
    "$MULTITASK_ROOT/StackCube-v1" \
    "$MULTITASK_ROOT/LiftPegUpright-v1" \
    "$MULTITASK_ROOT/PegInsertionSide-v1"