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

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

DEVICE="${DEVICE:-cuda:0}"
mkdir -p runs logs

python3 -m rrog.cli run \
  --task ogbg-molhiv --view gin --compute classic \
  --epochs 1 --hidden 32 --bs 64 --seed 991 --device "${DEVICE}" \
  --max_train_batches 2 --max_eval_batches 2

python3 -m rrog.cli run \
  --task ogbg-molhiv --view gin --compute fixed-rrog \
  --epochs 1 --hidden 32 --bs 64 --T 1 --n_sup 2 --seed 992 --device "${DEVICE}" \
  --max_train_batches 2 --max_eval_batches 2