| """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) | |