From 5e3e85003b021718289b7b4dc9930ad1bcaf7a95 Mon Sep 17 00:00:00 2001 From: Mariam Zakaria <123750992+mariam851@users.noreply.github.com> Date: Thu, 1 Jan 2026 19:13:37 +0200 Subject: [PATCH] Fix compatibility for make_forge and load_boston in sklearn 1.2+ --- mglearn/datasets.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/mglearn/datasets.py b/mglearn/datasets.py index 6d29925..6645621 100644 --- a/mglearn/datasets.py +++ b/mglearn/datasets.py @@ -32,12 +32,17 @@ def load_boston(): from sklearn.datasets import load_boston return load_boston() except ImportError: - pass - data_url = "http://lib.stat.cmu.edu/datasets/boston" - raw_df = pd.read_csv(data_url, sep="\s+", skiprows=22, header=None) - data = np.hstack([raw_df.values[::2, :], raw_df.values[1::2, :2]]) - target = raw_df.values[1::2, 2] - return Bunch(data=data, target=target) + data_url = "http://lib.stat.cmu.edu/datasets/boston" + raw_df = pd.read_csv(data_url, sep=r"\s+", skiprows=22, header=None) + data = np.hstack([raw_df.values[::2, :], raw_df.values[1::2, :2]]) + target = raw_df.values[1::2, 2] + + feature_names = ['CRIM', 'ZN', 'INDUS', 'CHAS', 'NOX', 'RM', 'AGE', 'DIS', 'RAD', 'TAX', 'PTRATIO', 'B', 'LSTAT'] + + return Bunch(data=data, + target=target, + feature_names=feature_names, + DESCR="Boston House Prices dataset (loaded from fallback URL)") def load_extended_boston():