Exception instead of generic assertion
Descriptive error message in BayesDiagNormal
Wrong type in fit_mleb
Evaluation of number of dims in BayesDiagNormal