modelgauge.prompt_pipeline should accept column headers as they appear in official prompt set files

Open
#901 1 comment 0 reactions 1 assignee View on GitHub

@rogthefrog is already working on this.

Since Mar 20, 2025.

Assessment

This issue has not been assessed yet.

Description

enhancement

The prompt_pipelines module expects prompt files to have PROMPT_CSV_INPUT_COLUMNS = ["UID", "Text"] as headers. The prompt set files use different column names, so you can't use prompt set files as inputs to modelgauge as-is.

Fix that.

Also document the acceptable column names.

Here's the prompt schema file Kurt created:
https://github.com/mlcommons/sugar/blob/prompts/prompts/schema.sql

Dominant language
Python
Stars
134
Forks
36
Avg merge
1d 11h
Merged PRs (30d)
17

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from mlcommons/modelbench

All issues in mlcommons/modelbench

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.