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,
        version="0.1.0",
        **kwargs,
    ):
        self.version = version
        self.arch_type = arch_type
        self.index_type = index_type
        self.novel = novel
        super().__init__(**kwargs)