Your dims don’t match the shape of your observed data. You should have as many dims as data dimensions.