from transformers import PretrainedConfig


class SmSubgroupClassifierConfig(PretrainedConfig):
    model_type = "sm_subgroup_classifier"

    def __init__(self, available_models=None, **kwargs):
        super().__init__(**kwargs)
        self.available_models = available_models or []