Update ALS-related annotations (#282)

third_party/3/pyspark/ml/recommendation.pyi

@@ -8,48 +8,55 @@ from pyspark.ml.wrapper import JavaEstimator, JavaModel
 from pyspark.ml.param.shared import *
 from pyspark.sql.dataframe import DataFrame
 
-class ALS(JavaEstimator[ALSModel], HasCheckpointInterval, HasMaxIter, HasPredictionCol, HasRegParam, HasSeed, JavaMLWritable, JavaMLReadable[ALS]):
+class _ALSModelParams(HasPredictionCol):
+    userCol: Param[str]
+    itemCol: Param[str]
+    coldStartStrategy: Param[str]
+    def getUserCol(self) -> str: ...
+    def getItemCol(self) -> str: ...
+    def getColdStartStrategy(self) -> str: ...
+
+class _ALSParams(_ALSModelParams, HasMaxIter, HasRegParam, HasCheckpointInterval, HasSeed):
     rank: Param[int]
     numUserBlocks: Param[int]
     numItemBlocks: Param[int]
     implicitPrefs: Param[bool]
     alpha: Param[float]
-    userCol: Param[str]
-    itemCol: Param[str]
     ratingCol: Param[str]
     nonnegative: Param[bool]
     intermediateStorageLevel: Param[str]
     finalStorageLevel: Param[str]
-    coldStrategy: Param[str]
+    def getRank(self) -> int: ...
+    def getNumUserBlocks(self) -> int: ...
+    def getNumItemBlocks(self) -> int: ...
+    def getImplicitPrefs(self) -> bool: ...
+    def getAlpha(self) -> float: ...
+    def getRatingCol(self) -> str: ...
+    def getNonnegative(self) -> bool: ...
+    def getIntermediateStorageLevel(self) -> str: ...
+    def getFinalStorageLevel(self) -> str: ...
+
+class ALS(JavaEstimator[ALSModel], _ALSParams, JavaMLWritable, JavaMLReadable[ALS]):
     def __init__(self, *, rank: int = ..., maxIter: int = ..., regParam: float = ..., numUserBlocks: int = ..., numItemBlocks: int = ..., implicitPrefs: bool = ..., alpha: float = ..., userCol: str = ..., itemCol: str = ..., seed: Optional[int] = ..., ratingCol: str = ..., nonnegative: bool = ..., checkpointInterval: int = ..., intermediateStorageLevel: str = ..., finalStorageLevel: str = ...,  coldStartStrategy: str = ...) -> None: ...
     def setParams(self, *, rank: int = ..., maxIter: int = ..., regParam: float = ..., numUserBlocks: int = ..., numItemBlocks: int = ..., implicitPrefs: bool = ..., alpha: float = ..., userCol: str = ..., itemCol: str = ..., seed: Optional[int] = ..., ratingCol: str = ..., nonnegative: bool = ..., checkpointInterval: int = ..., intermediateStorageLevel: str = ..., finalStorageLevel: str = ...,  coldStartStrategy: str = ...) -> ALS: ...
     def setRank(self, value: int) -> ALS: ...
-    def getRank(self) -> int: ...
     def setNumUserBlocks(self, value: int) -> ALS: ...
-    def getNumUserBlocks(self) -> int: ...
     def setNumItemBlocks(self, value: int) -> ALS: ...
-    def getNumItemBlocks(self) -> int: ...
     def setNumBlocks(self, value: int) -> ALS: ...
     def setImplicitPrefs(self, value: bool) -> ALS: ...
-    def getImplicitPrefs(self) -> bool: ...
     def setAlpha(self, value: float) -> ALS: ...
-    def getAlpha(self) -> float: ...
     def setUserCol(self, value: str) -> ALS: ...
-    def getUserCol(self) -> str: ...
     def setItemCol(self, value: str) -> ALS: ...
-    def getItemCol(self) -> str: ...
     def setRatingCol(self, value: str) -> ALS: ...
-    def getRatingCol(self) -> str: ...
     def setNonnegative(self, value: bool) -> ALS: ...
-    def getNonnegative(self) -> bool: ...
     def setIntermediateStorageLevel(self, value: str) -> ALS: ...
-    def getIntermediateStorageLevel(self) -> str: ...
     def setFinalStorageLevel(self, value: str) -> ALS: ...
-    def getFinalStorageLevel(self) -> str: ...
     def setColdStartStrategy(self, value: str) -> ALS: ...
-    def getColdStartStrategy(self) -> str: ...
 
-class ALSModel(JavaModel, JavaMLWritable, JavaMLReadable[ALSModel]):
+class ALSModel(JavaModel, _ALSModelParams, JavaMLWritable, JavaMLReadable[ALSModel]):
+    def setUserCol(self, value: str) -> ALSModel: ...
+    def setItemCol(self, value: str) -> ALSModel: ...
+    def setColdStartStrategy(self, value: str) -> ALSModel: ...
     @property
     def rank(self) -> int: ...
     @property