|
18 | 18 | SingleNodeMasterConfigEntry, |
19 | 19 | MultiNodeMasterConfigEntry, |
20 | 20 | ChangelogEntry, |
| 21 | + ChangelogMatrixEntry, |
21 | 22 | validate_matrix_entry, |
22 | 23 | validate_master_config, |
23 | 24 | validate_runner_config, |
@@ -1400,6 +1401,113 @@ def test_scenario_type_must_be_nonempty_and_supported(self, scenario_type): |
1400 | 1401 | }) |
1401 | 1402 |
|
1402 | 1403 |
|
| 1404 | +# ============================================================================= |
| 1405 | +# Test ChangelogMatrixEntry |
| 1406 | +# ============================================================================= |
| 1407 | + |
| 1408 | +@pytest.fixture |
| 1409 | +def valid_agentic_eval_row(): |
| 1410 | + """Agentic SWE-bench eval row as emitted by generate_sweep_configs --evals-only.""" |
| 1411 | + return { |
| 1412 | + "image": "vllm/vllm-openai:nightly-dev-x86_64-cu13.0.1-904e4ec", |
| 1413 | + "model": "deepseek-ai/DeepSeek-V4-Pro", |
| 1414 | + "model-prefix": "dsv4", |
| 1415 | + "precision": "fp4", |
| 1416 | + "framework": "vllm", |
| 1417 | + "runner": "cluster:b300-nv", |
| 1418 | + "tp": 8, |
| 1419 | + "pp": 1, |
| 1420 | + "dcp-size": 1, |
| 1421 | + "pcp-size": 1, |
| 1422 | + "ep": 8, |
| 1423 | + "dp-attn": True, |
| 1424 | + "spec-decoding": "mtp", |
| 1425 | + "conc": 224, |
| 1426 | + "kv-offloading": "none", |
| 1427 | + "router": {"name": "vllm-router", "version": "0.1.14"}, |
| 1428 | + "total-cpu-dram-gb": 0, |
| 1429 | + "duration": 3600, |
| 1430 | + "exp-name": "dsv4_tp8_conc224_kvnone_spec-mtp", |
| 1431 | + "scenario-type": "agentic-coding", |
| 1432 | + "run-eval": True, |
| 1433 | + "eval-only": True, |
| 1434 | + } |
| 1435 | + |
| 1436 | + |
| 1437 | +@pytest.fixture |
| 1438 | +def valid_changelog_metadata(): |
| 1439 | + return { |
| 1440 | + "base_ref": "base", |
| 1441 | + "head_ref": "head", |
| 1442 | + "entries": [{ |
| 1443 | + "config-keys": ["test-config"], |
| 1444 | + "description": ["Test entry"], |
| 1445 | + "pr-link": "https://github.com/SemiAnalysisAI/InferenceX/pull/1", |
| 1446 | + }], |
| 1447 | + } |
| 1448 | + |
| 1449 | + |
| 1450 | +class TestChangelogMatrixEntry: |
| 1451 | + """Tests for the final search-space contract consumed by run-sweep.yml.""" |
| 1452 | + |
| 1453 | + def test_agentic_eval_rows_validate_in_agentic_evals( |
| 1454 | + self, valid_agentic_eval_row, valid_changelog_metadata, |
| 1455 | + ): |
| 1456 | + entry = ChangelogMatrixEntry.model_validate({ |
| 1457 | + "agentic_evals": [valid_agentic_eval_row], |
| 1458 | + "changelog_metadata": valid_changelog_metadata, |
| 1459 | + }) |
| 1460 | + |
| 1461 | + assert entry.agentic_evals[0].scenario_type == "agentic-coding" |
| 1462 | + assert entry.agentic_evals[0].run_eval is True |
| 1463 | + assert entry.agentic_evals[0].eval_only is True |
| 1464 | + |
| 1465 | + def test_evals_bucket_rejects_agentic_rows( |
| 1466 | + self, valid_agentic_eval_row, valid_changelog_metadata, |
| 1467 | + ): |
| 1468 | + """The `evals` bucket is dispatched with fixed-seq-len inputs |
| 1469 | + (isl/osl/max-model-len), so agentic rows must never validate there.""" |
| 1470 | + with pytest.raises(ValueError): |
| 1471 | + ChangelogMatrixEntry.model_validate({ |
| 1472 | + "evals": [valid_agentic_eval_row], |
| 1473 | + "changelog_metadata": valid_changelog_metadata, |
| 1474 | + }) |
| 1475 | + |
| 1476 | + def test_evals_bucket_accepts_fixed_seq_len_rows( |
| 1477 | + self, valid_single_node_matrix_entry, valid_changelog_metadata, |
| 1478 | + ): |
| 1479 | + eval_row = { |
| 1480 | + **valid_single_node_matrix_entry, |
| 1481 | + "run-eval": True, |
| 1482 | + "eval-only": True, |
| 1483 | + } |
| 1484 | + entry = ChangelogMatrixEntry.model_validate({ |
| 1485 | + "evals": [eval_row], |
| 1486 | + "changelog_metadata": valid_changelog_metadata, |
| 1487 | + }) |
| 1488 | + |
| 1489 | + assert entry.evals[0].run_eval is True |
| 1490 | + assert entry.evals[0].eval_only is True |
| 1491 | + |
| 1492 | + def test_agentic_benchmark_rows_dump_without_eval_flags( |
| 1493 | + self, valid_agentic_eval_row, valid_changelog_metadata, |
| 1494 | + ): |
| 1495 | + """Benchmark rows omit run-eval/eval-only, and exclude_none must keep |
| 1496 | + them out of the dump so the dispatched row shape is unchanged.""" |
| 1497 | + benchmark_row = { |
| 1498 | + k: v for k, v in valid_agentic_eval_row.items() |
| 1499 | + if k not in ("run-eval", "eval-only") |
| 1500 | + } |
| 1501 | + entry = ChangelogMatrixEntry.model_validate({ |
| 1502 | + "single_node": {"agentic": [benchmark_row]}, |
| 1503 | + "changelog_metadata": valid_changelog_metadata, |
| 1504 | + }) |
| 1505 | + dumped = entry.model_dump(by_alias=True, exclude_none=True) |
| 1506 | + |
| 1507 | + assert "run-eval" not in dumped["single_node"]["agentic"][0] |
| 1508 | + assert "eval-only" not in dumped["single_node"]["agentic"][0] |
| 1509 | + |
| 1510 | + |
1403 | 1511 | # ============================================================================= |
1404 | 1512 | # Test load_config_files |
1405 | 1513 | # ============================================================================= |
|
0 commit comments