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

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "${ROOT_DIR}"
export PYTHONPATH="${ROOT_DIR}:${PYTHONPATH:-}"

ZINC_DEVICE="${ZINC_DEVICE:-cuda:0}"
OGB_DEVICE="${OGB_DEVICE:-cuda:1}"
OGB_TASK="${OGB_TASK:-ogbg-molhiv}"
ZINC_EPOCHS="${ZINC_EPOCHS:-200}"
OGB_EPOCHS="${OGB_EPOCHS:-100}"
SEED="${SEED:-0}"

mkdir -p runs logs

echo "[launch] ZINC-cycle56 on ${ZINC_DEVICE}"
DEVICE="${ZINC_DEVICE}" EPOCHS="${ZINC_EPOCHS}" SEED="${SEED}" \
  ./scripts/run_zinc_cycle56_full.sh > "logs/zinc_cycle56_${SEED}.log" 2>&1 &
zinc_pid=$!

echo "[launch] ${OGB_TASK} on ${OGB_DEVICE}"
TASK="${OGB_TASK}" DEVICE="${OGB_DEVICE}" EPOCHS="${OGB_EPOCHS}" SEED="${SEED}" \
  ./scripts/run_ogb_mol_task_full.sh > "logs/${OGB_TASK}_${SEED}.log" 2>&1 &
ogb_pid=$!

echo "[pids] zinc=${zinc_pid} ogb=${ogb_pid}"
wait "${zinc_pid}"
wait "${ogb_pid}"

echo "[done] collecting summaries"
./scripts/collect_results.sh