Skip to content

Commit 8a28ea8

Browse files
committed
fix test failures
1 parent c438f78 commit 8a28ea8

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

sklearn/feature_selection/from_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33

44
import numpy as np
55

6-
from ..base import (TransformerMixin, BaseEstimator, clone, SelectorMixin,
6+
from .base import SelectorMixin
7+
from ..base import (TransformerMixin, BaseEstimator, clone,
78
MetaEstimatorMixin)
89
from ..externals import six
910

sklearn/utils/estimator_checks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1446,7 +1446,7 @@ def fit(self, X, y):
14461446
if name in ('FeatureUnion', 'Pipeline'):
14471447
e = estimator([('clf', T())])
14481448

1449-
elif name in ('GridSearchCV' 'RandomizedSearchCV'):
1449+
elif name in ('GridSearchCV', 'RandomizedSearchCV', 'SelectFromModel'):
14501450
return
14511451

14521452
else:

0 commit comments

Comments
 (0)