BUG: Make sure to distinguish 1d row vs 1d obs in predict