Add real YOLO train acceptance

This commit is contained in:
2026-06-30 23:33:43 +08:00
parent 5055084788
commit fb96c96d8b
10 changed files with 270 additions and 16 deletions

View File

@@ -3,7 +3,12 @@ from __future__ import annotations
import time
from typing import Any
from .acceptance import latest_acceptance_report, latest_deep_acceptance_report, latest_real_acceptance_report
from .acceptance import (
latest_acceptance_report,
latest_deep_acceptance_report,
latest_real_acceptance_report,
latest_real_train_acceptance_report,
)
from .catalog import get_catalog
from .coverage import get_coverage_report
from .modules.dataset.service import list_uploaded_datasets
@@ -157,6 +162,7 @@ def get_capability_matrix() -> dict[str, Any]:
gpus = get_gpus()
acceptance = latest_acceptance_report()
real_acceptance = latest_real_acceptance_report()
real_train_acceptance = latest_real_train_acceptance_report()
deep_acceptance = latest_deep_acceptance_report()
all_tasks = catalog["task_types"]
@@ -277,6 +283,12 @@ def get_capability_matrix() -> dict[str, Any]:
"passed": bool(real_acceptance.get("passed")),
"detail": real_acceptance.get("run_id", "not run"),
},
{
"id": "real_train_acceptance",
"label": "真实短训练验收",
"passed": bool(real_train_acceptance.get("passed")),
"detail": real_train_acceptance.get("run_id", "not run"),
},
{
"id": "weights_manifest",
"label": "权重清单",
@@ -302,6 +314,7 @@ def get_capability_matrix() -> dict[str, Any]:
"gpus_available": bool(gpus.get("available")),
"acceptance_passed": bool(acceptance.get("passed")),
"real_acceptance_passed": bool(real_acceptance.get("passed")),
"real_train_acceptance_passed": bool(real_train_acceptance.get("passed")),
"deep_acceptance_passed": bool(deep_acceptance.get("passed")),
},
"requirements": requirements,