From 76b5837418c6a6bab7eabc5fb8296942adbc3048 Mon Sep 17 00:00:00 2001 From: Naoto Mizuno Date: Wed, 27 Sep 2023 13:25:08 +0900 Subject: [PATCH] Replaces California housing dataset with iris dataset --- .../lightgbm_tuner_tests/test_optimize.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/integration_tests/lightgbm_tuner_tests/test_optimize.py b/tests/integration_tests/lightgbm_tuner_tests/test_optimize.py index f2c66a1691..f24b67e142 100644 --- a/tests/integration_tests/lightgbm_tuner_tests/test_optimize.py +++ b/tests/integration_tests/lightgbm_tuner_tests/test_optimize.py @@ -727,9 +727,9 @@ def test_optuna_callback(self) -> None: assert callback_mock.call_count == 10 def test_tune_best_score_reproducibility(self) -> None: - california = sklearn.datasets.fetch_california_housing() + iris = sklearn.datasets.load_iris() X_trainval, X_test, y_trainval, y_test = train_test_split( - california.data, california.target, random_state=0 + iris.data, iris.target, random_state=0 ) train = lgb.Dataset(X_trainval, y_trainval) @@ -1058,9 +1058,9 @@ def test_get_best_booster_with_error(self) -> None: tuner3.get_best_booster() def test_tune_best_score_reproducibility(self) -> None: - california = sklearn.datasets.fetch_california_housing() + iris = sklearn.datasets.load_iris() X_trainval, X_test, y_trainval, y_test = train_test_split( - california.data, california.target, random_state=0 + iris.data, iris.target, random_state=0 ) train = lgb.Dataset(X_trainval, y_trainval)