Correct validateAndTransformSchema

sharp-pixel · sharp-pixel · commit fa5e3f94801b · 2017-08-17T23:48:58.000+02:00
diff --git a/mllib/src/main/scala/org/apache/spark/ml/clustering/GaussianMixture.scala b/mllib/src/main/scala/org/apache/spark/ml/clustering/GaussianMixture.scala
@@ -64,8 +64,8 @@ private[clustering] trait GaussianMixtureParams extends Params with HasMaxIter w
    */
   protected def validateAndTransformSchema(schema: StructType): StructType = {
     SchemaUtils.checkColumnType(schema, $(featuresCol), new VectorUDT)
-    SchemaUtils.appendColumn(schema, $(predictionCol), IntegerType)
-    SchemaUtils.appendColumn(schema, $(probabilityCol), new VectorUDT)
+    val schema_temp = SchemaUtils.appendColumn(schema, $(predictionCol), IntegerType)
+    SchemaUtils.appendColumn(schema_temp, $(probabilityCol), new VectorUDT)
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -64,8 +64,8 @@ private[clustering] trait GaussianMixtureParams extends Params with HasMaxIter w`
`64`	`64`	`*/`
`65`	`65`	`protected def validateAndTransformSchema(schema: StructType): StructType = {`
`66`	`66`	`SchemaUtils.checkColumnType(schema, $(featuresCol), new VectorUDT)`
`67`		`- SchemaUtils.appendColumn(schema, $(predictionCol), IntegerType)`
`68`		`- SchemaUtils.appendColumn(schema, $(probabilityCol), new VectorUDT)`
	`67`	`+ val schema_temp = SchemaUtils.appendColumn(schema, $(predictionCol), IntegerType)`
	`68`	`+ SchemaUtils.appendColumn(schema_temp, $(probabilityCol), new VectorUDT)`
`69`	`69`	`}`
`70`	`70`	`}`
`71`	`71`