Skip to content

Commit 1dffeee

Browse files
committed
[SPARK-5726] [MLLIB] Pass style checks.
1 parent e436896 commit 1dffeee

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mllib/src/main/scala/org/apache/spark/ml/feature/ElementwiseProduct.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@ class ElementwiseProduct extends UnaryTransformer[Vector, Vector, ElementwisePro
3333

3434
/** the vector to multiply with input vectors */
3535
val scalingVec : Param[Vector] = new Param(this, "scalingVector", "vector for hadamard product")
36-
def setScalingVec(value: Vector) = set(scalingVec, value)
37-
def getScalingVec: Vector = get(scalingVec)
36+
def setScalingVec(value: Vector): this.type = set(scalingVec, value)
37+
def getScalingVec: Vector = getOrDefault(scalingVec)
3838

3939
override protected def createTransformFunc(paramMap: ParamMap): Vector => Vector = {
4040
val elemScaler = new feature.ElementwiseProduct(paramMap(scalingVec))

0 commit comments

Comments
 (0)