"""SnakeStateEncoder: the Snake board (8 channels, see task.py) -> core tokens.

A thin, Snake-specific configuration of the generic GridStateEncoder: the Snake
featurisation lives in `task.encode_grid`, the learnable part is generic.
"""
from minijev.encoders.grid import GridStateEncoder

from .task import N_CHANNELS


class SnakeStateEncoder(GridStateEncoder):
    def __init__(self, d: int, in_channels: int = N_CHANNELS, **kw):
        super().__init__(d, in_channels=in_channels, **kw)