Skip to content

Add Inspect backends for MTBench#12

Merged
adamhrin-amd merged 1 commit into
mainfrom
fix-mtbench-endpoint-inspect
Jul 17, 2026
Merged

Add Inspect backends for MTBench#12
adamhrin-amd merged 1 commit into
mainfrom
fix-mtbench-endpoint-inspect

Conversation

@adamhrin-amd

@adamhrin-amd adamhrin-amd commented Jul 16, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Add endpoint-backed Inspect tasks for English mt_bench_inspect and Finnish mt_bench_fi_inspect.
  • Register English alongside the existing native mt_bench task, while keeping native mt_bench on the existing flowaicom/Flow-Judge-v0.1 vLLM judge path.
  • Register Finnish as a built-in task via mt_bench_fi/main.py, backed by LumiOpen/mtbench_multi, subset fi, split test.
  • Run both Inspect tasks as true two-turn MTBench tasks: generate turn 1, append turn 2, generate turn 2, then score turn_1 and turn_2 separately.
  • Use an OpenAI-compatible scorer endpoint for judging via SCORER_MODEL_BASE_URL, SCORER_MODEL_PATH, and VLLM_API_KEY.
  • Preserve MTBench reference answers when present. Finnish scoring reuses the existing Flow-Judge MTBench prompt templates and adds only a small Finnish-content note.
  • Validate scorer endpoint configuration in the Inspect solver before generation, while creating the scorer model lazily so registry loading and native task imports do not require scorer env vars.

Verification

  • ruff check src/lighteval/tasks/tasks/mt_bench/main_inspect.py src/lighteval/tasks/tasks/mt_bench_fi/main.py src/lighteval/tasks/tasks/mt_bench_fi/main_inspect.py src/lighteval/main_inspect.py
  • python3 -m py_compile src/lighteval/tasks/tasks/mt_bench/main_inspect.py src/lighteval/tasks/tasks/mt_bench_fi/main.py src/lighteval/tasks/tasks/mt_bench_fi/main_inspect.py src/lighteval/main_inspect.py
  • TensorWave Qwen3.5-4B smokes completed successfully with /shared_silo/scratch/models/Qwen3.5-4B as target and judge:
    • English mt_bench_inspect: job 38686, dashboard eval 927.
    • Finnish mt_bench_fi_inspect: job 38687, dashboard eval 928.
  • The smokes used runtime --reasoning-tags set to [('<think>', '</think>'), ('', '</think>')]; raw logs observed thinking traces before stripping, and scratch assertions confirmed scorer answers and judge prompts were tag-free for both turns.

Notes

  • The successful TensorWave smokes used the current TensorBase MTBench wrapper shape: no --custom-tasks, compatible vLLM020 packages, and scorer endpoint env vars exported by the wrapper.
  • The Finnish dataset loaded as 80 test rows from LumiOpen/mtbench_multi / fi during the smoke.

@adamhrin-amd
adamhrin-amd force-pushed the fix-mtbench-endpoint-inspect branch from 18ccdc5 to d546bc2 Compare July 16, 2026 13:57
@adamhrin-amd adamhrin-amd changed the title Make MTBench endpoint-judge tasks inspect-compatible Allow MTBench judge model override Jul 16, 2026
@adamhrin-amd
adamhrin-amd force-pushed the fix-mtbench-endpoint-inspect branch from d546bc2 to 8e9821a Compare July 17, 2026 07:26
@adamhrin-amd adamhrin-amd changed the title Allow MTBench judge model override Add Inspect backend for MTBench Jul 17, 2026
@adamhrin-amd
adamhrin-amd force-pushed the fix-mtbench-endpoint-inspect branch from 8e9821a to d8bce56 Compare July 17, 2026 08:33
@adamhrin-amd adamhrin-amd changed the title Add Inspect backend for MTBench Add Inspect backends for MTBench Jul 17, 2026
@adamhrin-amd
adamhrin-amd force-pushed the fix-mtbench-endpoint-inspect branch from d8bce56 to 1262e8d Compare July 17, 2026 09:46
@adamhrin-amd
adamhrin-amd requested review from chaanim and ezosa July 17, 2026 11:34

@ezosa ezosa left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Approved. But we might want to add a language checking step to MTBench-FI so that only responses classified as Finnish by a language identifier (e.g. langdetect) will be judged by the LLM. From previous experience, LLMs are not always consistent when asked to identify the language of the response.

@adamhrin-amd

Copy link
Copy Markdown
Collaborator Author

Okay! Merging this one and will open a new work for the lang detection

@adamhrin-amd
adamhrin-amd merged commit 80b5fe9 into main Jul 17, 2026
4 of 5 checks passed
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