Skip to content

Commit 197a3bd

Browse files
committed
CLN appease ruff/linting
1 parent 379cfdf commit 197a3bd

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

tests/python_package_test/test_sklearn.py

+6-10
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
from sklearn.utils.validation import check_is_fitted
2323

2424
import lightgbm as lgb
25+
from lightgbm.basic import LGBMDeprecationWarning
2526
from lightgbm.compat import (
2627
DASK_INSTALLED,
2728
DATATABLE_INSTALLED,
@@ -31,7 +32,6 @@
3132
pd_DataFrame,
3233
pd_Series,
3334
)
34-
from lightgbm.basic import LGBMDeprecationWarning
3535

3636
from .utils import (
3737
assert_silent,
@@ -2049,9 +2049,7 @@ def test_classifier_fit_detects_classes_every_time():
20492049
def 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():
20612059
def 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

Comments
 (0)