And here you should also use something from `sklearn.utils.validation`, either `array2d`, `atleast2d_or_csr/csc` or `check_arrays` I guess.  This should ensure proper error messages.  I don't think we have any util that explicitly prevents >2d though, hmm.