[SPARK-5726] [MLLIB] Pass style checks.

chenghao-intel · May 7, 2015 · 1dffeee · 1dffeee
1 parent e436896
commit 1dffeee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/ElementwiseProduct.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/ElementwiseProduct.scala
@@ -33,8 +33,8 @@ class ElementwiseProduct extends UnaryTransformer[Vector, Vector, ElementwisePro
 
   /** the vector to multiply with input vectors */
   val scalingVec : Param[Vector] = new Param(this, "scalingVector", "vector for hadamard product")
-  def setScalingVec(value: Vector) = set(scalingVec, value)
-  def getScalingVec: Vector = get(scalingVec)
+  def setScalingVec(value: Vector): this.type = set(scalingVec, value)
+  def getScalingVec: Vector = getOrDefault(scalingVec)
 
   override protected def createTransformFunc(paramMap: ParamMap): Vector => Vector = {
     val elemScaler = new feature.ElementwiseProduct(paramMap(scalingVec))