saridormi's picture
Add metrics computation for CMG task
cdf268e
raw
history blame
395 Bytes
from typing import Dict, Optional
from .base_task_metrics import BaseTaskMetrics
from .commit_message_generation import CMGMetrics
METRICS: Dict[str, Optional[BaseTaskMetrics]] = {
"commit_message_generation": CMGMetrics(),
"bug_localization": None,
"module_to_text": None,
"library_usage": None,
"project_code_completion": None,
"bug_localization_build_logs": None,
}