"git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "fedf3971832201849c2bcdac334289cc97e46217"
Unverified Commit 1a294c87 authored by Zhuyi Xue's avatar Zhuyi Xue Committed by GitHub
Browse files

[ci][python] apply isort to tests/python_package_test/test_dask.py #3958 (#3979)

parent a630da3e
......@@ -8,8 +8,10 @@ from itertools import groupby
from os import getenv
from sys import platform
import lightgbm as lgb
import pytest
import lightgbm as lgb
if not platform.startswith('linux'):
pytest.skip('lightgbm.dask is currently supported in Linux environments', allow_module_level=True)
if not lgb.compat.DASK_INSTALLED:
......@@ -21,16 +23,15 @@ import dask.dataframe as dd
import joblib
import numpy as np
import pandas as pd
from scipy.stats import spearmanr
from dask.array.utils import assert_eq
from dask.distributed import default_client, Client, LocalCluster, wait
from dask.distributed import Client, LocalCluster, default_client, wait
from distributed.utils_test import client, cluster_fixture, gen_cluster, loop
from scipy.sparse import csr_matrix
from scipy.stats import spearmanr
from sklearn.datasets import make_blobs, make_regression
from .utils import make_ranking
# time, in seconds, to wait for the Dask client to close. Used to avoid teardown errors
# see https://distributed.dask.org/en/latest/api.html#distributed.Client.close
CLIENT_CLOSE_TIMEOUT = 120
......
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