from transformers import PretrainedConfig, AutoConfig
from typing import List


class BioNExtExtractorConfig(PretrainedConfig):
    model_type = "relation-novelty-extractor"

    def __init__(
        self,
        arch_type = "mha",
        index_type = "both",
        novel = True,
        tokenizer_special_tokens = ['[s1]','[e1]', '[s2]','[e2]' ],
        update_vocab = None,
        version="0.1.1",
        **kwargs,
    ):
        self.version = version
        self.arch_type = arch_type
        self.index_type = index_type
        self.novel = novel
        self.tokenizer_special_tokens = tokenizer_special_tokens
        self.update_vocab = update_vocab
        super().__init__(**kwargs)