Skip to content

Commit

Permalink
[ci][python] apply isort to tests/python_package_test/test_dask.py mi…
Browse files Browse the repository at this point in the history
  • Loading branch information
zyxue committed Feb 14, 2021
1 parent fd094d2 commit 7848ad8
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions tests/python_package_test/test_dask.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,10 @@
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:
Expand All @@ -21,16 +23,15 @@
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
Expand Down

0 comments on commit 7848ad8

Please sign in to comment.