Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[pyspark] fix a type hint with old pyspark release (#9079)

Browse Source
This commit is contained in:
Bobby Wang 2023-04-24 20:04:14 +08:00 committed by GitHub
parent d237378452
commit 339f21e1bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python-package/xgboost/spark/params.py
View File

@ -1,4 +1,6 @@
"""Xgboost pyspark integration submodule for params.""" """Xgboost pyspark integration submodule for params."""
from typing import Dict
# pylint: disable=too-few-public-methods # pylint: disable=too-few-public-methods
from pyspark.ml.param import TypeConverters from pyspark.ml.param import TypeConverters
from pyspark.ml.param.shared import Param, Params from pyspark.ml.param.shared import Param, Params
@ -11,7 +13,7 @@ class HasArbitraryParamsDict(Params):
input. input.
""" """
arbitrary_params_dict: Param[dict] = Param( arbitrary_params_dict: "Param[Dict]" = Param(
Params._dummy(), Params._dummy(),
"arbitrary_params_dict", "arbitrary_params_dict",
"arbitrary_params_dict This parameter holds all of the additional parameters which are " "arbitrary_params_dict This parameter holds all of the additional parameters which are "