from transformers import PretrainedConfig, AutoConfig
from typing import List


class BioNExtTaggerConfig(PretrainedConfig):
    model_type = "crf-tagger"

    def __init__(
        self,
        augmentation = "unk",
        context_size = 64,
        percentage_tags = 0.2,
        p_augmentation = 0.5,
        crf_reduction = "mean",
        version="0.1.2",
        **kwargs,
    ):
        self.version = version
        self.augmentation = augmentation
        self.context_size = context_size
        self.percentage_tags = percentage_tags
        self.p_augmentation = p_augmentation
        self.crf_reduction = crf_reduction
        super().__init__(**kwargs)