code/minijev/keys.py

"""Model input names per state-encoder kind (torch-free, shared by training, runtime and ONNX)."""
STATE_INPUTS = {"grid": "state_grid", "text": "state_ids", "log": "state_log_ids", "numeric": "state_x"}