Nevertheless, both in training and testing the world state is observable, and the alignment and the text are conditioned on the state during inference.
there was no need to model the distribution of the world state.