Fix tests on numpy master

lesteve · lesteve · commit 8fb95cb9b642 · 2017-02-14T15:16:16.000+01:00
numpy.apply_along_axis has changed behaviour when the function passed
in returns a 2d array
diff --git a/sklearn/gaussian_process/kernels.py b/sklearn/gaussian_process/kernels.py
@@ -1852,7 +1852,7 @@ def diag(self, X):
             Diagonal of kernel k(X, X)
         """
         # We have to fall back to slow way of computing diagonal
-        return np.apply_along_axis(self, 1, X)[:, 0]
+        return np.apply_along_axis(lambda arr: self(arr).ravel(), 1, X)[:, 0]
 
     def is_stationary(self):
         """Returns whether the kernel is stationary. """
