lint.

dmlc · Nov 6, 2024 · 05881a5 · 05881a5
1 parent 55b116a
commit 05881a5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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: