2222from sklearn .utils .validation import check_is_fitted
2323
2424import lightgbm as lgb
25+ from lightgbm .basic import LGBMDeprecationWarning
2526from lightgbm .compat import (
2627 DASK_INSTALLED ,
2728 DATATABLE_INSTALLED ,
3132 pd_DataFrame ,
3233 pd_Series ,
3334)
34- from lightgbm .basic import LGBMDeprecationWarning
3535
3636from .utils import (
3737 assert_silent ,
@@ -2049,9 +2049,7 @@ def test_classifier_fit_detects_classes_every_time():
20492049def test_eval_set_deprecation ():
20502050 """Test use of eval_set raises deprecation warning."""
20512051 X , y = make_synthetic_regression (n_samples = 10 )
2052- X_train , X_test , y_train , y_test = train_test_split (
2053- X , y , test_size = 0.5 , random_state = 42
2054- )
2052+ X_train , X_test , y_train , y_test = train_test_split (X , y , test_size = 0.5 , random_state = 42 )
20552053 gbm = lgb .LGBMRegressor ()
20562054 msg = "The argument 'eval_set' is deprecated.*"
20572055 with pytest .warns (LGBMDeprecationWarning , match = msg ):
@@ -2061,10 +2059,8 @@ def test_eval_set_deprecation():
20612059def test_eval_X_eval_y_eval_set_equivalence ():
20622060 """Test that eval_X and eval_y are equivalent to eval_set."""
20632061 X , y = make_synthetic_regression ()
2064- X_train , X_test , y_train , y_test = train_test_split (
2065- X , y , test_size = 0.25 , random_state = 42
2066- )
2067- cbs = [lgb .early_stopping (2 )]
2062+ X_train , X_test , y_train , y_test = train_test_split (X , y , test_size = 0.25 , random_state = 42 )
2063+ cbs = [lgb .early_stopping (2 )]
20682064 gbm1 = lgb .LGBMRegressor ()
20692065 gbm1 .fit (X_train , y_train , eval_set = (X_test , y_test ), callbacks = cbs )
20702066 gbm2 = lgb .LGBMRegressor ()
@@ -2073,8 +2069,8 @@ def test_eval_X_eval_y_eval_set_equivalence():
20732069
20742070 # 2 evaluation sets
20752071 n = X_test .shape [0 ]
2076- X_test1 , X_test2 = X_test [:n // 2 ], X_test [n // 2 :]
2077- y_test1 , y_test2 = y_test [:n // 2 ], y_test [n // 2 :]
2072+ X_test1 , X_test2 = X_test [: n // 2 ], X_test [n // 2 :]
2073+ y_test1 , y_test2 = y_test [: n // 2 ], y_test [n // 2 :]
20782074 gbm1 = lgb .LGBMRegressor ()
20792075 gbm1 .fit (X_train , y_train , eval_set = [(X_test1 , y_test1 ), (X_test2 , y_test2 )], callbacks = cbs )
20802076 gbm2 = lgb .LGBMRegressor ()
0 commit comments