"""图像级标签形式的共用定义（模型与数据集都要用到，放这里避免互相 import）。"""

from __future__ import annotations

MULTILABEL = "multilabel"
MULTICLASS = "multiclass"
LABEL_MODES = (MULTILABEL, MULTICLASS)


def check_label_mode(label_mode: str) -> str:
    if label_mode not in LABEL_MODES:
        raise ValueError(f"未知 label_mode: {label_mode}，可选 {LABEL_MODES}")
    return label_mode