Simulator should produce observation array of shape (1,x,...) instead of (x,...) - to be explicit about the fact that it is a single observation. Otherwise, we get an error inside the learner.

[yegortk]