"""``transformers`` configuration of encoder-tvsd; the model itself is in ``modeling.py``.

``config.json`` describes every variant: the shared fields, ``default_variant``, and ``variants``
(each variant's ``members``, ``areas`` and ``scores``). There is deliberately no ``variant``
attribute: ``AutoModel.from_pretrained(..., variant="tiny")`` must reach the model class, which picks
the variant and its weight file.
"""

from transformers import PretrainedConfig


class EncoderTVSDConfig(PretrainedConfig):
    model_type = "encoder_tvsd"

    def __init__(
        self,
        default_variant: str = "",
        variants: dict | None = None,
        monkeys: list | None = None,
        electrode_index: dict | None = None,
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.default_variant = default_variant
        self.variants = variants or {}
        self.monkeys = monkeys or []
        self.electrode_index = electrode_index or {}