FIX call check_array in Nystroem fit & transform

ogrisel · ogrisel · commit 090ad08ed4ad · 2015-02-16T17:56:14.000+01:00
diff --git a/sklearn/kernel_approximation.py b/sklearn/kernel_approximation.py
@@ -437,10 +437,8 @@ def fit(self, X, y=None):
         X : array-like, shape=(n_samples, n_feature)
             Training data.
         """
-
+        X = check_array(X, accept_sparse='csr')
         rnd = check_random_state(self.random_state)
-        if not sp.issparse(X):
-            X = np.asarray(X)
         n_samples = X.shape[0]
 
         # get basis vectors
@@ -487,6 +485,7 @@ def transform(self, X):
             Transformed data.
         """
         check_is_fitted(self, 'components_')
+        X = check_array(X, accept_sparse='csr')
 
         kernel_params = self._get_kernel_params()
         embedded = pairwise_kernels(X, self.components_,
