diff --git a/third_party/3/pyspark/ml/clustering.pyi b/third_party/3/pyspark/ml/clustering.pyi index 50eb5b8e..c0b3563d 100644 --- a/third_party/3/pyspark/ml/clustering.pyi +++ b/third_party/3/pyspark/ml/clustering.pyi @@ -55,6 +55,7 @@ class _GaussianMixtureParams( HasTol, HasAggregationDepth, HasWeightCol, + HasBlockSize, ): k: Param[int] def getK(self) -> int: ... @@ -97,7 +98,8 @@ class GaussianMixture( maxIter: int = ..., seed: Optional[int] = ..., aggregationDepth: int = ..., - weightCol: Optional[str] = ... + weightCol: Optional[str] = ..., + blockSize: int = ... ) -> None: ... def setParams( self, @@ -110,7 +112,8 @@ class GaussianMixture( maxIter: int = ..., seed: Optional[int] = ..., aggregationDepth: int = ..., - weightCol: Optional[str] = ... + weightCol: Optional[str] = ..., + blockSize: int = ... ) -> GaussianMixture: ... def setK(self, value: int) -> GaussianMixture: ... def setMaxIter(self, value: int) -> GaussianMixture: ... @@ -121,6 +124,7 @@ class GaussianMixture( def setSeed(self, value: int) -> GaussianMixture: ... def setTol(self, value: float) -> GaussianMixture: ... def setAggregationDepth(self, value: int) -> GaussianMixture: ... + def setBlockSize(self, value: int) -> GaussianMixture: ... class GaussianMixtureSummary(ClusteringSummary): @property