from dataclasses import dataclass


@dataclass
class HFCol:
    input_ids: str = "input_ids"
    text: str = "text"
    id: str = "id"
    ner_tags: str = "ner_tags"
    text_token_indices: str = "text_token_indices"
    tokens: str = "tokens"
    bpe_tokens: str = "bpe_tokens"
    gold_spans: str = "gold_spans"
    labels: str = "labels"


@dataclass
class SpanKey:
    label: str = "label"
    start: str = "start"
    end: str = "end"
