[doc] Add missing document for pyspark ranker. [skip ci] (#8692)

2023-01-18 07:52:18 +08:00
parent 78396f8a6e
commit 175986b739
3 changed files with 16 additions and 5 deletions
--- a/python-package/xgboost/spark/init.py
+++ b/python-package/xgboost/spark/init.py
@@ -1,5 +1,4 @@
-"""PySpark XGBoost integration interface
-"""
+"""PySpark XGBoost integration interface"""

 try:
    import pyspark
@@ -10,6 +9,7 @@ from .estimator import (
    SparkXGBClassifier,
    SparkXGBClassifierModel,
    SparkXGBRanker,
+    SparkXGBRankerModel,
    SparkXGBRegressor,
    SparkXGBRegressorModel,
 )
@@ -20,4 +20,5 @@ __all__ = [
    "SparkXGBRegressor",
    "SparkXGBRegressorModel",
    "SparkXGBRanker",
+    "SparkXGBRankerModel",
 ]