| set -euo pipefail | |
| if [ "$#" -lt 2 ]; then | |
| echo "Usage: $0 CHECKPOINT REPLAY_DIR [extra evaluate args]" >&2 | |
| exit 1 | |
| fi | |
| checkpoint="$1" | |
| replay_dir="$2" | |
| shift 2 | |
| export PYTHONPATH="${PYTHONPATH:-}:$(pwd)/src" | |
| python3.11 src/evaluate.py \ | |
| planner \ | |
| --config configs/fourrooms_world_model.yaml \ | |
| --checkpoint "${checkpoint}" \ | |
| "$@" | |
| python3.11 src/evaluate.py \ | |
| world-model \ | |
| --config configs/fourrooms_world_model.yaml \ | |
| --checkpoint "${checkpoint}" \ | |
| --replay-dir "${replay_dir}" \ | |
| "$@" | |