Add flatten argument to ANOVATest/ChisqTest/FValueTest (#481)

Resolves #410
zero323 · Aug 29, 2020 · 2bee703 · 2bee703
1 parent 154e14e
commit 2bee703
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/third_party/3/pyspark/ml/stat.pyi b/third_party/3/pyspark/ml/stat.pyi
@@ -29,7 +29,9 @@ from py4j.java_gateway import JavaObject  # type: ignore[import]
 
 class ChiSquareTest:
     @staticmethod
-    def test(dataset: DataFrame, featuresCol: str, labelCol: str) -> DataFrame: ...
+    def test(
+        dataset: DataFrame, featuresCol: str, labelCol: str, flatten: bool = ...
+    ) -> DataFrame: ...
 
 class Correlation:
     @staticmethod
@@ -78,8 +80,12 @@ class MultivariateGaussian:
 
 class ANOVATest:
     @staticmethod
-    def test(dataset: DataFrame, featuresCol: str, labelCol: str) -> DataFrame: ...
+    def test(
+        dataset: DataFrame, featuresCol: str, labelCol: str, flatten: bool = ...
+    ) -> DataFrame: ...
 
 class FValueTest:
     @staticmethod
-    def test(dataset: DataFrame, featuresCol: str, labelCol: str) -> DataFrame: ...
+    def test(
+        dataset: DataFrame, featuresCol: str, labelCol: str, flatten: bool = ...
+    ) -> DataFrame: ...