make `fitted_flocker` return in the shape of `obs`

[jsocolar]

Fixed on the more-models branch! This should be a big unlock for all sorts of prediction tasks!