Unverified Commit 4007b34f authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[dask] fix Dask import order (#3788)

parent a2927664
......@@ -12,6 +12,8 @@ from urllib.parse import urlparse
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 dask.distributed import Client, default_client, get_worker, wait
from .basic import _LIB, _safe_call
from .sklearn import LGBMClassifier, LGBMRegressor
import scipy.sparse as ss
logger = logging.getLogger(__name__)
......
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment