def predict_state(posterior, state_transition):
    """The prior given the previous posterior density and a transition
    matrix indicating the state at the next time step.
    """
    return state_transition @ posterior
