Skip to content

Commit eb3ff73

Browse files
committed
fix import mismatch
1 parent e458cd5 commit eb3ff73

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

mlxtend/feature_selection/tests/test_sequential_feature_selector_feature_groups.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,16 +3,15 @@
33
# Author: Sebastian Raschka <sebastianraschka.com>
44
#
55
# License: BSD 3 clause
6-
from mlxtend.data import boston_housing_data
76
import numpy as np
8-
from numpy import nan
97
from numpy.testing import assert_almost_equal
108
from sklearn.datasets import load_iris
119
from sklearn.ensemble import RandomForestClassifier
1210
from sklearn.linear_model import LinearRegression
1311
from sklearn.metrics import roc_auc_score
1412
from sklearn.neighbors import KNeighborsClassifier
1513

14+
from mlxtend.data import boston_housing_data
1615
from mlxtend.feature_selection import SequentialFeatureSelector as SFS
1716
from mlxtend.utils import assert_raises
1817

@@ -163,8 +162,7 @@ def test_keyboard_interrupt():
163162

164163

165164
def test_max_feature_subset_best():
166-
boston = load_boston()
167-
X, y = boston.data, boston.target
165+
X, y = boston_housing_data
168166
lr = LinearRegression()
169167

170168
sfs = SFS(
@@ -189,8 +187,7 @@ def test_max_feature_subset_best():
189187

190188

191189
def test_max_feature_subset_parsimonious():
192-
boston = load_boston()
193-
X, y = boston.data, boston.target
190+
X, y = boston_housing_data
194191
lr = LinearRegression()
195192

196193
sfs = SFS(

0 commit comments

Comments
 (0)