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