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