#!/bin/bash
#SBATCH --job-name=dovla_ms_multi_rgb
#SBATCH --account=def-yalda_cpu
#SBATCH --partition=cpubase_bycore_b1
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=8
#SBATCH --mem=24G
#SBATCH --time=03:00:00
#SBATCH --array=0-4%2
#SBATCH --output=outputs/hpc/logs/%x_%A_%a.out
#SBATCH --error=outputs/hpc/logs/%x_%A_%a.err

set -euo pipefail

PROJECT_DIR="${PROJECT_DIR:-$SLURM_SUBMIT_DIR}"
MULTITASK_OUT_ROOT="${MULTITASK_OUT_ROOT:?Set MULTITASK_OUT_ROOT}"

case "${SLURM_ARRAY_TASK_ID:-0}" in
  0) ENV_ID="PushCube-v1" ;;
  1) ENV_ID="PullCube-v1" ;;
  2) ENV_ID="StackCube-v1" ;;
  3) ENV_ID="LiftPegUpright-v1" ;;
  4) ENV_ID="PegInsertionSide-v1" ;;
  *) echo "unsupported array index" >&2; exit 2 ;;
esac

export PROJECT_DIR
export DATASET="$MULTITASK_OUT_ROOT/$ENV_ID"
export IMAGE_QUALITY="${IMAGE_QUALITY:-85}"
export SEED="${SEED:-0}"

exec bash "$PROJECT_DIR/scripts/slurm/render_maniskill_observations.sbatch"