[backport] [doc] Add missing document for pyspark ranker. (#8692) (#8990)

2023-03-29 12:02:51 +08:00
parent f5f03dfb61
commit 365da0b8f4
3 changed files with 21 additions and 10 deletions
--- a/python-package/xgboost/spark/init.py
+++ b/python-package/xgboost/spark/init.py
@@ -1,6 +1,5 @@
 # type: ignore
-"""PySpark XGBoost integration interface
-"""
+"""PySpark XGBoost integration interface"""

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