Skip to content

Fix ARC evaluation label mapping for numeric choice IDs - #2

Merged
abcd1927 merged 2 commits into
sapientinc:mainfrom
grp06:fix-arc-numeric-labels
May 27, 2026
Merged

abcd1927 merged 2 commits into
sapientinc:mainfrom
grp06:fix-arc-numeric-labels

Conversation

@grp06

@grp06 grp06 commented May 19, 2026

Copy link
Copy Markdown
Contributor

Summary

  • Map ARC answer keys through choices.label instead of ASCII arithmetic.
  • Hide ARC row conversion behind a named adapter helper.
  • Add focused unit tests for numeric labels, letter labels, formatting, scoring, and missing labels.

Tests

  • python -m compileall -q evaluation/benchmarks.py tests/test_benchmarks.py
  • python -m pytest tests/test_benchmarks.py -q

Note

  • The optional live ARC smoke evaluation was not run because it requires a checkpoint and GPU.

@abcd1927

Copy link
Copy Markdown
Collaborator

Thanks for catching this @grp06 — confirmed the bug with HF dataset (22/1172 ARC-Challenge test rows have numeric answerKeys).

For this codebase I'd like to keep the change minimal — would you mind dropping the helper extraction and the test file, and just changing the lambda to:

row_to_doc_fn=lambda r: MCQDoc(r['question'], r['choices']['text'], r['choices']['label'].index(r['answerKey']))

We don't have a test harness set up here yet, so the stubs in tests/test_benchmarks.py would be the only test in the repo and it'd be a maintenance burden until we add a proper one. Happy to merge once it's down to the one-line change.

@abcd1927
abcd1927 merged commit 0c52b88 into sapientinc:main May 27, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants