Repository navigation
Commit 41de354
authored
Add AgentBoard BabyAI environment (#32)
## Summary
Add a self-contained AgentBoard BabyAI environment using the native
Verifiers 0.4 taskset and environment APIs.
- Bundle the text-game runtime, prompt, and 112 annotated episodes; no
AgentBoard checkout is needed.
- Preserve the existing deterministic 84/28 train/eval split across 28
task families, progress reward, action parsing, and grounding/success
metrics.
- Include a portable Prime Inference evaluation configuration and
concise setup instructions.
- Include upstream attribution and the separate runtime/prompt (Apache
2.0) and annotation (GPL 2.0) licenses.
Published environment: [bhoy1/agentboard-babyai, version
0.3.0](https://app.primeintellect.ai/dashboard/environments/bhoy1/agentboard-babyai).
## Verification
- Scoped pre-commit checks: Ruff lint/format, Markdown formatting/lint,
and Git LFS checks all passed.
- Repository package checks: 2 passed, including package build.
- Existing external regression suite: 23 passed and 1 optional upstream
comparison skipped. Includes 38 golden trajectory replays, all 112
initial observations, turn-limit behavior, concurrent rollout isolation,
and cleanup.
- Real Prime Inference evaluation from this contribution: Qwen3.5-9B, 1
held-out task with 2 rollouts, thinking disabled, 20 turns, 1,024-token
output budget. Both completed without harness errors, with reward 0.25
and stop condition `max_turns`.
The contribution's [evaluation
run](https://app.primeintellect.ai/dashboard/evaluations/savi3ix7zcjlua3x26nxxh7u)
was created.
<!-- Macroscope's pull request summary starts here -->
<!-- Macroscope will only edit the content between these invisible
markers, and the markers themselves will not be visible in the GitHub
rendered markdown. -->
<!-- If you delete either of the start / end markers from your PR's
description, Macroscope will append its summary at the bottom of the
description. -->
> [!NOTE]
> ### Add AgentBoard BabyAI environment package with text simulator and
eval config
> - Adds a new `agentboard-babyai` package with a
Gymnasium/MiniGrid-based text simulator (`BabyAI` in
[agentboard_babyai_runtime.py](https://github.com/PrimeIntellect-ai/residency-environments/pull/32/files#diff-be425817e801ab23a50f90676c99504ff1476095da3d7a460d535ff253c99181))
that validates high-level actions, tracks progress reward, and reports
success, failure, or step-limit termination
> - Adds a Verifiers integration
([agentboard_babyai.py](https://github.com/PrimeIntellect-ai/residency-environments/pull/32/files#diff-cfd10d39b442bf92beb1f2b3a6c64379b762914f1dfb8a6f2b87f70de52d522b))
that builds prompts from the packaged JSON asset, parses model replies
into single normalized actions, and drives multi-turn rollouts with
metrics for success, grounding, turns, and invalid actions
> - Supports `all`, `train_3_per_subtask`, and `eval_1_per_subtask`
splits with a configurable base seed (default 1234)
> - Includes an eval config
([eval.toml](https://github.com/PrimeIntellect-ai/residency-environments/pull/32/files#diff-650fda196fe27f64843882f04187d9520ba76efb843a7f3d84c2f6099ba6e808))
selecting Qwen/Qwen3.5-9B with thinking disabled, and packages task
data, prompt assets, and upstream AgentBoard provenance docs
> - Risk: the simulator reads MiniGrid internals via `env.unwrapped` and
expresses the 2D cross-product test as a scalar determinant; the
ordered-door verifiers (`_BeforeDoors`, `_AfterDoors`) changed failure
handling, so check those if port parity matters
>
> <!-- Macroscope's review summary starts here -->
>
> <sup><a href="https://app.macroscope.com">Macroscope</a> summarized
7a8d4a3.</sup>
> <!-- Macroscope's review summary ends here -->
>
<!-- Macroscope's pull request summary ends here -->1 parent 3cf818a commit 41de354
11 files changed
Lines changed: 1970 additions & 0 deletions
File tree
- configs/agentboard-babyai
- environments/agentboard-babyai
- assets
- licenses
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
0 commit comments