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

output_dir="${MINIDREAMER_OUTPUT_DIR:-artifacts/world_model_experiment}"

export PYTHONPATH="${PYTHONPATH:-}:$(pwd)/src"
python3.11 src/train_world_model.py \
  --config configs/fourrooms_world_model.yaml \
  --output-dir "${output_dir}" \
  "$@"