Specify src path for isort. (#8867)

tests/ci_build/lint_python.py

@@ -3,12 +3,15 @@ import os
 import subprocess
 import sys
 from multiprocessing import Pool, cpu_count
-from typing import Dict, Tuple
+from typing import Dict, Optional, Tuple
 
 from pylint import epylint
 from test_utils import PY_PACKAGE, ROOT, cd, print_time, record_time
 
 CURDIR = os.path.normpath(os.path.abspath(os.path.dirname(__file__)))
+SRCPATH = os.path.normpath(
+    os.path.join(CURDIR, os.path.pardir, os.path.pardir, "python-package")
+)
 
 
 @record_time
@@ -29,7 +32,7 @@ Please run the following command on your machine to address the formatting error
 
 @record_time
 def run_isort(rel_path: str) -> bool:
-    cmd = ["isort", "--check", "--profile=black", rel_path]
+    cmd = ["isort", f"--src={SRCPATH}", "--check", "--profile=black", rel_path]
     ret = subprocess.run(cmd).returncode
     if ret != 0:
         subprocess.run(["isort", "--version"])

tests/python-gpu/test_gpu_data_iterator.py

@@ -2,6 +2,7 @@ import sys
 
 import pytest
 from hypothesis import given, settings, strategies
+
 from xgboost.testing import no_cupy
 
 sys.path.append("tests/python")

tests/python-gpu/test_gpu_eval_metrics.py

@@ -1,10 +1,10 @@
 import sys
 
 import pytest
-from xgboost.testing.metrics import check_quantile_error
 
 import xgboost
 from xgboost import testing as tm
+from xgboost.testing.metrics import check_quantile_error
 
 sys.path.append("tests/python")
 import test_eval_metrics as test_em  # noqa

tests/python-gpu/test_gpu_prediction.py

@@ -3,10 +3,10 @@ import sys
 import numpy as np
 import pytest
 from hypothesis import assume, given, settings, strategies
-from xgboost.compat import PANDAS_INSTALLED
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.compat import PANDAS_INSTALLED
 
 if PANDAS_INSTALLED:
     from hypothesis.extra.pandas import column, data_frames, range_indexes

tests/python-gpu/test_gpu_updaters.py

@@ -4,11 +4,11 @@ from typing import Any, Dict
 import numpy as np
 import pytest
 from hypothesis import assume, given, note, settings, strategies
-from xgboost.testing.params import cat_parameter_strategy, hist_parameter_strategy
-from xgboost.testing.updater import check_init_estimation, check_quantile_loss
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.testing.params import cat_parameter_strategy, hist_parameter_strategy
+from xgboost.testing.updater import check_init_estimation, check_quantile_loss
 
 sys.path.append("tests/python")
 import test_updaters as test_up

tests/python/test_data_iterator.py

@@ -4,11 +4,11 @@ import numpy as np
 import pytest
 from hypothesis import given, settings, strategies
 from scipy.sparse import csr_matrix
-from xgboost.data import SingleBatchInternalIter as SingleBatch
-from xgboost.testing import IteratorForTest, make_batches, non_increasing
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.data import SingleBatchInternalIter as SingleBatch
+from xgboost.testing import IteratorForTest, make_batches, non_increasing
 
 pytestmark = tm.timeout(30)
 

tests/python/test_dmatrix.py

@@ -6,10 +6,10 @@ import pytest
 import scipy.sparse
 from hypothesis import given, settings, strategies
 from scipy.sparse import csr_matrix, rand
-from xgboost.testing.data import np_dtypes
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.testing.data import np_dtypes
 
 rng = np.random.RandomState(1)
 

tests/python/test_early_stopping.py

@@ -1,9 +1,9 @@
 import numpy as np
 import pytest
-from xgboost.testing.updater import get_basescore
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.testing.updater import get_basescore
 
 rng = np.random.RandomState(1994)
 

tests/python/test_eval_metrics.py

@@ -1,9 +1,9 @@
 import numpy as np
 import pytest
-from xgboost.testing.metrics import check_quantile_error
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.testing.metrics import check_quantile_error
 
 rng = np.random.RandomState(1337)
 

tests/python/test_predict.py

@@ -5,11 +5,11 @@ import numpy as np
 import pandas as pd
 import pytest
 from scipy import sparse
-from xgboost.testing.data import np_dtypes, pd_dtypes
-from xgboost.testing.shared import validate_leaf_output
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.testing.data import np_dtypes, pd_dtypes
+from xgboost.testing.shared import validate_leaf_output
 
 
 def run_threaded_predict(X, rows, predict_func):

tests/python/test_quantile_dmatrix.py

@@ -4,6 +4,8 @@ import numpy as np
 import pytest
 from hypothesis import given, settings, strategies
 from scipy import sparse
+
+import xgboost as xgb
 from xgboost.testing import (
     IteratorForTest,
     make_batches,
@@ -15,8 +17,6 @@ from xgboost.testing import (
 )
 from xgboost.testing.data import np_dtypes
 
-import xgboost as xgb
-
 
 class TestQuantileDMatrix:
     def test_basic(self) -> None:

tests/python/test_updaters.py

@@ -5,6 +5,9 @@ from typing import Any, Dict, List
 import numpy as np
 import pytest
 from hypothesis import given, note, settings, strategies
+
+import xgboost as xgb
+from xgboost import testing as tm
 from xgboost.testing.params import (
     cat_parameter_strategy,
     exact_parameter_strategy,
@@ -12,9 +15,6 @@ from xgboost.testing.params import (
 )
 from xgboost.testing.updater import check_init_estimation, check_quantile_loss
 
-import xgboost as xgb
-from xgboost import testing as tm
-
 
 def train_result(param, dmat, num_rounds):
     result = {}

tests/python/test_with_pandas.py

@@ -3,10 +3,10 @@ from typing import Type
 import numpy as np
 import pytest
 from test_dmatrix import set_base_margin_info
-from xgboost.testing.data import pd_arrow_dtypes, pd_dtypes
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.testing.data import pd_arrow_dtypes, pd_dtypes
 
 try:
     import pandas as pd

tests/python/test_with_sklearn.py

@@ -8,11 +8,11 @@ from typing import Callable, Optional
 import numpy as np
 import pytest
 from sklearn.utils.estimator_checks import parametrize_with_checks
-from xgboost.testing.shared import get_feature_weights, validate_data_initialization
-from xgboost.testing.updater import get_basescore
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.testing.shared import get_feature_weights, validate_data_initialization
+from xgboost.testing.updater import get_basescore
 
 rng = np.random.RandomState(1994)
 pytestmark = [pytest.mark.skipif(**tm.no_sklearn()), tm.timeout(30)]

tests/test_distributed/test_federated/test_federated.py

@@ -3,9 +3,8 @@ import multiprocessing
 import sys
 import time
 
-import xgboost.federated
-
 import xgboost as xgb
+import xgboost.federated
 
 SERVER_KEY = 'server-key.pem'
 SERVER_CERT = 'server-cert.pem'

tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py

@@ -10,10 +10,10 @@ import numpy as np
 import pytest
 from hypothesis import given, note, settings, strategies
 from hypothesis._settings import duration
-from xgboost.testing.params import hist_parameter_strategy
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.testing.params import hist_parameter_strategy
 
 pytestmark = [
     pytest.mark.skipif(**tm.no_dask()),
@@ -42,9 +42,9 @@ try:
     from dask import array as da
     from dask.distributed import Client
     from dask_cuda import LocalCUDACluster
-    from xgboost.testing.dask import check_init_estimation
 
     from xgboost import dask as dxgb
+    from xgboost.testing.dask import check_init_estimation
 except ImportError:
     pass
 

tests/test_distributed/test_gpu_with_spark/test_gpu_spark.py

@@ -12,6 +12,7 @@ pytestmark = pytest.mark.skipif(**tm.no_spark())
 from pyspark.ml.linalg import Vectors
 from pyspark.ml.tuning import CrossValidator, ParamGridBuilder
 from pyspark.sql import SparkSession
+
 from xgboost.spark import SparkXGBClassifier, SparkXGBRegressor
 
 gpu_discovery_script_path = "tests/test_distributed/test_gpu_with_spark/discover_gpu.sh"

tests/test_distributed/test_with_dask/test_with_dask.py

@@ -21,6 +21,9 @@ import scipy
 import sklearn
 from hypothesis import HealthCheck, given, note, settings
 from sklearn.datasets import make_classification, make_regression
+
+import xgboost as xgb
+from xgboost import testing as tm
 from xgboost.data import _is_cudf_df
 from xgboost.testing.params import hist_parameter_strategy
 from xgboost.testing.shared import (
@@ -29,9 +32,6 @@ from xgboost.testing.shared import (
     validate_leaf_output,
 )
 
-import xgboost as xgb
-from xgboost import testing as tm
-
 pytestmark = [tm.timeout(60), pytest.mark.skipif(**tm.no_dask())]
 
 import dask
@@ -39,6 +39,7 @@ import dask.array as da
 import dask.dataframe as dd
 from distributed import Client, LocalCluster
 from toolz import sliding_window  # dependency of dask
+
 from xgboost.dask import DaskDMatrix
 from xgboost.testing.dask import check_init_estimation
 

tests/test_distributed/test_with_spark/test_data.py

@@ -8,6 +8,7 @@ from xgboost import testing as tm
 
 pytestmark = [pytest.mark.skipif(**tm.no_spark())]
 
+from xgboost import DMatrix, QuantileDMatrix
 from xgboost.spark.data import (
     _read_csr_matrix_from_unwrapped_spark_vec,
     alias,
@@ -15,8 +16,6 @@ from xgboost.spark.data import (
     stack_series,
 )
 
-from xgboost import DMatrix, QuantileDMatrix
-
 
 def test_stack() -> None:
     a = pd.DataFrame({"a": [[1, 2], [3, 4]]})

tests/test_distributed/test_with_spark/test_spark_local.py

@@ -8,10 +8,10 @@ from typing import Generator, Sequence, Type
 
 import numpy as np
 import pytest
-from xgboost.spark.data import pred_contribs
 
 import xgboost as xgb
 from xgboost import testing as tm
+from xgboost.spark.data import pred_contribs
 
 pytestmark = [tm.timeout(60), pytest.mark.skipif(**tm.no_spark())]
 
@@ -23,6 +23,8 @@ from pyspark.ml.linalg import Vectors
 from pyspark.ml.tuning import CrossValidator, ParamGridBuilder
 from pyspark.sql import SparkSession
 from pyspark.sql import functions as spark_sql_func
+
+from xgboost import XGBClassifier, XGBModel, XGBRegressor
 from xgboost.spark import (
     SparkXGBClassifier,
     SparkXGBClassifierModel,
@@ -32,8 +34,6 @@ from xgboost.spark import (
 )
 from xgboost.spark.core import _non_booster_params
 
-from xgboost import XGBClassifier, XGBModel, XGBRegressor
-
 from .utils import SparkTestCase
 
 logging.getLogger("py4j").setLevel(logging.INFO)

tests/test_distributed/test_with_spark/test_spark_local_cluster.py

@@ -11,6 +11,7 @@ from xgboost import testing as tm
 pytestmark = pytest.mark.skipif(**tm.no_spark())
 
 from pyspark.ml.linalg import Vectors
+
 from xgboost.spark import SparkXGBClassifier, SparkXGBRegressor
 from xgboost.spark.utils import _get_max_num_concurrent_tasks
 

tests/test_distributed/test_with_spark/utils.py

@@ -13,6 +13,7 @@ from xgboost import testing as tm
 pytestmark = [pytest.mark.skipif(**tm.no_spark())]
 
 from pyspark.sql import SparkSession
+
 from xgboost.spark.utils import _get_default_params_from_func