cy0307's picture
Update final Qwen model scripts
627e5d7 verified
Raw
History Blame
1.22 kB
"""Canonical reader-facing names for public task-suite artifacts."""
from __future__ import annotations
from typing import Any
TASK_DISPLAY_NAMES = {
"timeline_action": "Action Recognition",
"timeline_subtask": "Procedure Step Recognition",
"transition_detection": "Action Boundary Detection",
"next_action": "Next-Action Prediction",
"hand_trajectory_forecast": "Hand Trajectory Forecasting",
"contact_prediction": "Contact State Prediction",
"object_relevance": "Object Relevance Prediction",
"caption_grounding": "Language Grounding",
"cross_modal_retrieval": "Cross-Modal Retrieval",
"modality_reconstruction": "Cross-Modal Reconstruction",
"temporal_order": "Temporal Order Verification",
"misalignment_detection": "Multimodal Synchronization Detection",
}
def task_display_name(task_id: str) -> str:
"""Return the public label while preserving unknown ids for debugging."""
return TASK_DISPLAY_NAMES.get(task_id, task_id.replace("_", " ").title())
def with_task_display(task_id: str, payload: dict[str, Any]) -> dict[str, Any]:
payload.setdefault("task", task_id)
payload["task_display_name"] = task_display_name(task_id)
return payload