"""Hugging Face configuration for TinyLunarPolicy (32-param linear LunarLander policy)."""
from transformers import PretrainedConfig


class TinyLunarConfig(PretrainedConfig):
    model_type = "tiny-lunar-policy"

    def __init__(self, obs_dim: int = 8, n_actions: int = 4, bias: bool = False, **kwargs):
        self.obs_dim = int(obs_dim)
        self.n_actions = int(n_actions)
        self.bias = bool(bias)
        super().__init__(**kwargs)