metrics/__init__.py

# metrics/__init__.py
"""
Central metrics entrypoint: import and expose all metric functions.
"""
from .core import compute_all_metrics_batch
from .bleu import compute_bleu_single, section_bleu, full_bleu, compute_bleu_single
from .bleurt import compute_bleurt_single, get_hf_bleurt
from .rouge import compute_rouge_single, get_hf_rouge
from .bertscore import compute_bertscore_single, BERT_FRIENDLY_TO_MODEL, BERT_MODEL_TO_FRIENDLY

__all__ = [
    "compute_all_metrics_batch",
    "compute_bleu_single",
    "compute_bleurt_single",
    "compute_rouge_single",
    "get_hf_bleurt",
    "get_hf_rouge",
    "compute_bertscore_single",
    "BERT_FRIENDLY_TO_MODEL",
    "BERT_MODEL_TO_FRIENDLY",
]