Contributing
Benchmarks
QUAIL-B is a benchmark of AI-SQL filter and join queries over six datasets at three scale factors (0.1, 0.5, 1.0). It has 30+ queries. We are actively adding queries and datasets.
| Family | Dataset | Queries |
|---|---|---|
| IMDB | stanfordnlp/imdb | 10 |
| BIO | BioDEX/BioDEX-Reactions | 4 |
| FEV | fever/fever | 10 |
| LEP | rmahari/LePaRD | 5 |
| AGENT | TIGER-Lab/SWE-Next-SFT-Trajectories | 2 |
| PRIV | mukund/PrivacyPolicies | 2 |
PRIV queries are optional and run at larger scale. Pass --privacy
to include them. The query count tracks the pinned quail_b version
in pyproject.toml.
Run a query
On your GPU:
uv run python -m quail.bench.quailb --sf 0.1 --only IMDB-4 \
--model qwen3-4b-fp8 --device h100-sxm \
--output-dir results/quailb/imdb-4On Modal, with the vLLM baselines:
uv run modal run --detach -m quail.bench.quailb_parallel \
--sf 0.1 --model qwen3-4b-fp8 --query IMDB-4 \
--output-dir /results/benchmarks/quailb \
2>&1 | tee results/benchmark/quailb.logOmit --only / --query to run all queries. Add --no-include-baselines
to run Quail alone on Modal. Modal results are saved to the
quail-results volume under /results/benchmarks/quailb/<run-id>/.
Generate a report
uv run quail-b report results/quailb/imdb-4For a Modal run, download it first:
uv run modal volume get quail-results \
"benchmarks/quailb/<run-id>/" "./results/<run-id>/"
uv run quail-b report "./results/<run-id>/quail"This writes report.md next to run.json. Reports use saved labels
and do not rerun inference.