Skip to content

Commit

Permalink
2 spaces instead of 1
Browse files Browse the repository at this point in the history
  • Loading branch information
BenFradet committed Jun 26, 2015
1 parent 1ddb452 commit 26f5a7a
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -339,7 +339,7 @@ class VectorIndexerModel private[ml] (
override def transform(dataset: DataFrame): DataFrame = {
transformSchema(dataset.schema, logging = true)
val newField = prepOutputField(dataset.schema)
val transformUDF = udf { (vector: Vector) => transformFunc(vector) }
val transformUDF = udf { (vector: Vector) => transformFunc(vector) }
val newCol = transformUDF(dataset($(inputCol)))
dataset.withColumn($(outputCol), newCol.as($(outputCol), newField.metadata))
}
Expand Down

0 comments on commit 26f5a7a

Please sign in to comment.