diff --git a/python-package/lightgbm/dask.py b/python-package/lightgbm/dask.py index fb8b06077e7..0e40e453433 100644 --- a/python-package/lightgbm/dask.py +++ b/python-package/lightgbm/dask.py @@ -12,6 +12,8 @@ import numpy as np import pandas as pd +import scipy.sparse as ss + from dask import array as da from dask import dataframe as dd from dask import delayed @@ -20,8 +22,6 @@ from .basic import _LIB, _safe_call from .sklearn import LGBMClassifier, LGBMRegressor -import scipy.sparse as ss - logger = logging.getLogger(__name__) diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py index 901584dafd9..1a454f6c6c8 100644 --- a/tests/python_package_test/test_dask.py +++ b/tests/python_package_test/test_dask.py @@ -109,7 +109,7 @@ def test_training_does_not_fail_on_port_conflicts(client): n_estimators=5, num_leaves=5 ) - for i in range(5): + for _ in range(5): dask_classifier.fit( X=dX, y=dy, @@ -204,7 +204,7 @@ def test_regressor_quantile(output, client, listen_port, alpha): def test_regressor_local_predict(client, listen_port): - X, y, w, dX, dy, dw = _create_data('regression', output='array') + X, y, _, dX, dy, dw = _create_data('regression', output='array') dask_regressor = dlgbm.DaskLGBMRegressor( local_listen_port=listen_port,