pybasicbayes with diagonalgaussian nan asserts, make obs plotting work with nans