diff --git a/third_party/3/pyspark/ml/param/__init__.pyi b/third_party/3/pyspark/ml/param/__init__.pyi index 2813b014..8c637f04 100644 --- a/third_party/3/pyspark/ml/param/__init__.pyi +++ b/third_party/3/pyspark/ml/param/__init__.pyi @@ -70,3 +70,4 @@ class Params(pyspark.ml.util.Identifiable): def extractParamMap(self, extra: Optional[pyspark.ml._typing.ParamMap] = ...) -> pyspark.ml._typing.ParamMap: ... def copy(self, extra: Optional[pyspark.ml._typing.ParamMap] = ...) -> Params: ... def set(self, param: Param, value: Any) -> None: ... + def clear(self, param: Param) -> None: ... diff --git a/third_party/3/pyspark/ml/wrapper.pyi b/third_party/3/pyspark/ml/wrapper.pyi index cef998ba..1b7d1d56 100644 --- a/third_party/3/pyspark/ml/wrapper.pyi +++ b/third_party/3/pyspark/ml/wrapper.pyi @@ -5,7 +5,7 @@ from typing import Any, Generic, Optional, Type, TypeVar from pyspark.ml._typing import P, T, JM, ParamMap from pyspark.ml import Estimator, Model, Transformer -from pyspark.ml.param import Params +from pyspark.ml.param import Param, Params from pyspark.ml.param.shared import HasFeaturesCol, HasLabelCol, HasPredictionCol xrange = range @@ -17,6 +17,7 @@ class JavaWrapper: class JavaParams(JavaWrapper, Params): __metaclass__: Type[abc.ABCMeta] def copy(self: P, extra: Optional[ParamMap] = ...) -> P: ... + def clear(self, param: Param) -> None: ... class JavaEstimator(JavaParams, Estimator[JM]): __metaclass__: Type[abc.ABCMeta]