Add Params.clear

zero323 · Jan 12, 2020 · 382744c · 382744c
1 parent e921766
commit 382744c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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]