small bugfix when using sgd for mle since sgd has no booked information and not the full belief state is given in the dataset
