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

WAIT_PID="${1:?usage: run_zinc_gine_after_pid.sh <pid>}"
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "${ROOT_DIR}"

echo "[wait] OGB queue pid=${WAIT_PID}"
while kill -0 "${WAIT_PID}" 2>/dev/null; do
  sleep 60
done

echo "[start] OGB queue exited; launching ZINC gine"
DEVICE="${DEVICE:-cuda:1}" EPOCHS="${EPOCHS:-200}" SEED="${SEED:-0}" ./rrog/run_zinc_gine.sh