Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 13 additions & 17 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -72,6 +72,8 @@ uv run python

**Download the files.** Run this example from the repository root. It downloads all five tables from [Hugging Face](https://huggingface.co/datasets/Andromede-AI/messier/tree/main) into the folders our loaders and analyses read:

> **Note:** Downloading all five tables requires more than 9 GB of disk space. The trajectories account for most of this size.

```python
from huggingface_hub import hf_hub_download

Expand Down Expand Up @@ -117,23 +119,23 @@ This example streams rows from Hugging Face. Change the filename to stream recor
Below we show several use cases and quick starts for using MESSIER.


Combine all five tables into one complete dataset containing the full information.
Collect all stored information for one task.

```python
from collections import defaultdict
import pandas as pd
from messier.io import data_path, load_jsonl

tables = {}
for name in ("tasks", "records", "verifiers", "trajectories", "classifications"):
rows_by_task = defaultdict(list)
for row in load_jsonl(data_path(f"{name}.jsonl", source="local")):
rows_by_task[(row["benchmark"], row["task_id"])].append(row)
task = next(load_jsonl(data_path("tasks.jsonl", source="local")))
key = (task["benchmark"], task["task_id"])

tables[name] = pd.Series(rows_by_task, dtype=object)
task_data = {}
for name in ("tasks", "records", "verifiers", "trajectories", "classifications"):
task_data[name] = [
row
for row in load_jsonl(data_path(f"{name}.jsonl", source="local"))
if (row["benchmark"], row["task_id"]) == key
]

dataset = pd.DataFrame(tables).rename_axis(["benchmark", "task_id"]).reset_index()
print(dataset.head())
print({name: len(rows) for name, rows in task_data.items()})
```

Compare agents within each benchmark using trials with a score.
Expand All @@ -158,12 +160,6 @@ uv run --group dev python analysis/frontier_progress/run.py
# Results: analysis/outputs/frontier_progress/
```

The script downloads and verifies the frozen raw sources, rebuilds the corpus, and runs validation tests.

```bash
bash scripts/update.sh
```

## License

MESSIER is licensed under [MIT](LICENSE).
Expand Down
4 changes: 2 additions & 2 deletions docs/data.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
| Term | Meaning |
| --- | --- |
| Model | The underlying language model |
| Scaffold | The system that turns model outputs into actions and parses environment observations |
| Scaffold | The system that turns model outputs into actions and parses environment observations (sometimes called a harness) |
| Agent | A model-scaffold pair. If a source reports a reasoning setting, such as `high`, we append it to the agent ID |
| Environment | The system with which an agent interacts, defining the states, actions, and observations available |
| Task | A given problem consisting of a string instruction (and an initial environment state) |
Expand All @@ -37,7 +37,7 @@ The release files use the same terminology as the paper. Names inside a metadata
| **records.jsonl** | One stored trial result, verifier result, or source summary |
| **trajectories.jsonl** | One available step-by-step execution trajectory (agent per task) |
| **classifications.jsonl** | One task's final SOC and NAICS labels, the labels proposed by each classification model, and adjudication details when the models disagree |
| **task_files/** | Original files available to agents for tasks that provide them |
| **task_files/** | Original files available to agents for tasks that provide them, sometimes omitted because their licenses do not permit redistribution |

For example, a trial with three separately reported verifier results produces four records: three verifier-result rows and one trial-result row.

Expand Down