Skip to content

Commit c138bfe

Browse files
committed
code fix
1 parent 244e57e commit c138bfe

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

mlxtend/feature_selection/tests/test_exhaustive_feature_selector.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
#
55
# License: BSD 3 clause
66

7+
from mlxtend.data import boston_housing_data
78
import numpy as np
89
import pandas as pd
910
from numpy.testing import assert_almost_equal
@@ -334,10 +335,8 @@ def test_fit_params():
334335

335336

336337
def test_regression():
337-
data_url = "http://lib.stat.cmu.edu/datasets/boston"
338-
raw_df = pd.read_csv(data_url, sep="\s+", skiprows=22, header=None)
339-
boston = raw_df.values[1::2, 2]
340-
X, y = boston.data[:, [1, 2, 6, 8, 12]], boston.target
338+
X, y = boston_housing_data()
339+
X = X[:, [1, 2, 6, 8, 12]]
341340
lr = LinearRegression()
342341
efs_r = EFS(
343342
lr,

0 commit comments

Comments
 (0)