src/evaluation/metrics.py

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,
}