Correct return type of Params.hasParam (#284)

zero323 · Jan 11, 2020 · 52de6cc · 52de6cc
1 parent 606464c
commit 52de6cc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/third_party/3/pyspark/ml/param/__init__.pyi b/third_party/3/pyspark/ml/param/__init__.pyi
@@ -53,16 +53,16 @@ class Params(pyspark.ml.util.Identifiable):
     @overload
     def isSet(self, param: str) -> bool: ...
     @overload
-    def isSet(self, param: Param) -> bool: ...
+    def isSet(self, param: Param[Any]) -> bool: ...
     @overload
     def hasDefault(self, param: str) -> bool: ...
     @overload
-    def hasDefault(self, param: Param) -> bool: ...
+    def hasDefault(self, param: Param[Any]) -> bool: ...
     @overload
     def isDefined(self, param: str) -> bool: ...
     @overload
-    def isDefined(self, param: Param) -> bool: ...
-    def hasParam(self, paramName: str) -> Param: ...
+    def isDefined(self, param: Param[Any]) -> bool: ...
+    def hasParam(self, paramName: str) -> bool: ...
     @overload
     def getOrDefault(self, param: str) -> Any: ...
     @overload