Update python-package/xgboost/spark/core.py

Co-authored-by: Jiaming Yuan <[email protected]>
wbo4958 · Sep 5, 2023 · accf7e1 · accf7e1
1 parent 5014580
commit accf7e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/xgboost/spark/core.py b/python-package/xgboost/spark/core.py
@@ -1200,7 +1200,7 @@ def _post_transform(self, dataset: DataFrame, pred_col: Column) -> DataFrame:
         return dataset
 
     def _gpu_transform(self) -> bool:
-        """If gpu is used to do the prediction, true to gpu prediction"""
+        """Whether GPU is used for prediction."""
 
         if _is_local(_get_spark_session().sparkContext):
             # if it's local model, we just use the internal "device"