From 05881a5e087f36c0cc5438c5ea52cc6205c7e7a4 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Thu, 7 Nov 2024 03:26:38 +0800 Subject: [PATCH] lint. --- python-package/xgboost/spark/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python-package/xgboost/spark/utils.py b/python-package/xgboost/spark/utils.py index afffd2c85d14..a8a2314272a6 100644 --- a/python-package/xgboost/spark/utils.py +++ b/python-package/xgboost/spark/utils.py @@ -14,12 +14,12 @@ from pyspark import BarrierTaskContext, SparkConf, SparkContext, SparkFiles, TaskContext from pyspark.sql.session import SparkSession -from xgboost import Booster, XGBModel -from xgboost.collective import CommunicatorContext as CCtx -from xgboost.tracker import RabitTracker - +from ..collective import CommunicatorContext as CCtx from ..collective import _Args as CollArgs from ..collective import _ArgVals as CollArgsVals +from ..core import Booster +from ..sklearn import XGBModel +from ..tracker import RabitTracker def get_class_name(cls: Type) -> str: