Skip to content

Commit 4cdfbc2

Browse files
committed
bench: consolidate AIME export to suites only; set num_runs=8 default
1 parent ddc9b74 commit 4cdfbc2

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

eval_protocol/benchmarks/run.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,8 @@ def _parse_args() -> argparse.Namespace:
3838
help="Limit rows: integer or 'all' for no limit (maps to EP_MAX_DATASET_ROWS)",
3939
)
4040
parser.add_argument("--num-runs", type=int, help="Override num_runs if provided")
41+
# Allow overriding reasoning effort explicitly (low/medium/high). If omitted, suite default is used.
42+
# Already mapped by --reasoning-effort above.
4143
return parser.parse_args()
4244

4345

eval_protocol/benchmarks/suites/aime25.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def aime2025_dataset_adapter(rows: List[Dict[str, Any]]) -> List[EvaluationRow]:
7474
rollout_processor=default_single_turn_rollout_processor,
7575
aggregation_method="mean",
7676
threshold_of_success=None,
77-
num_runs=1,
77+
num_runs=8,
7878
max_dataset_rows=2,
7979
max_concurrent_rollouts=4,
8080
mode="pointwise",

examples/aime2025_chat_completion/tests/test_evaluation.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
default_single_turn_rollout_processor,
77
)
88
from eval_protocol.pytest.evaluation_test import evaluation_test
9-
from eval_protocol.benchmarks.registry import export_benchmark
109

1110
from examples.aime2025_chat_completion.main import _extract_boxed_text, _normalize_to_int_or_none
1211

@@ -56,7 +55,6 @@ def aime2025_dataset_adapter(rows: List[Dict[str, Any]]) -> List[EvaluationRow]:
5655
return converted
5756

5857

59-
@export_benchmark("aime25_low")
6058
@evaluation_test(
6159
model=["fireworks_ai/accounts/fireworks/models/gpt-oss-120b"],
6260
input_dataset=[

0 commit comments

Comments
 (0)