ENH Checks n_features_in_ after fitting in random_projection (#19541)

thomasjpfan · web-flow · commit 12db86ee1ba6 · 2021-02-25T14:59:27.000+01:00
diff --git a/sklearn/random_projection.py b/sklearn/random_projection.py
@@ -38,7 +38,7 @@
 from .utils import check_random_state
 from .utils.extmath import safe_sparse_dot
 from .utils.random import sample_without_replacement
-from .utils.validation import check_array, check_is_fitted
+from .utils.validation import check_is_fitted
 from .utils.validation import _deprecate_positional_args
 from .exceptions import DataDimensionalityWarning
 
@@ -402,9 +402,8 @@ def transform(self, X):
         X_new : {ndarray, sparse matrix} of shape (n_samples, n_components)
             Projected array.
         """
-        X = check_array(X, accept_sparse=['csr', 'csc'])
-
         check_is_fitted(self)
+        X = self._validate_data(X, accept_sparse=['csr', 'csc'], reset=False)
 
         if X.shape[1] != self.components_.shape[1]:
             raise ValueError(
diff --git a/sklearn/tests/test_common.py b/sklearn/tests/test_common.py
@@ -272,7 +272,6 @@ def test_search_cv(estimator, check, request):
     'multiclass',
     'multioutput',
     'pipeline',
-    'random_projection',
 }
 
 N_FEATURES_IN_AFTER_FIT_ESTIMATORS = [

Original file line number	Diff line number	Diff line change
`@@ -272,7 +272,6 @@ def test_search_cv(estimator, check, request):`
`272`	`272`	`'multiclass',`
`273`	`273`	`'multioutput',`
`274`	`274`	`'pipeline',`
`275`		`- 'random_projection',`
`276`	`275`	`}`
`277`	`276`
`278`	`277`	`N_FEATURES_IN_AFTER_FIT_ESTIMATORS = [`