ODIn/spark-instrumented-optimizer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

[MLlib] UnaryTransformer nullability should not depend on PrimitiveType.

Browse source 
Author: Reynold Xin <rxin@databricks.com>

Closes #5644 from rxin/mllib-nullable and squashes the following commits:

a727e5b [Reynold Xin] [MLlib] UnaryTransformer nullability should not depend on primitive types.
This commit is contained in:
Reynold Xin 2015-04-22 21:35:12 -07:00
parent b69c4f9b2e
commit 1b85e08509
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
mllib/src/main/scala/org/apache/spark/ml/Transformer.scala
View file

@ -94,7 +94,7 @@ private[ml] abstract class UnaryTransformer[IN, OUT, T <: UnaryTransformer[IN, O
throw new IllegalArgumentException(s"Output column ${map(outputCol)} already exists.") throw new IllegalArgumentException(s"Output column ${map(outputCol)} already exists.")
} }
val outputFields = schema.fields :+ val outputFields = schema.fields :+
StructField(map(outputCol), outputDataType, !outputDataType.isPrimitive) StructField(map(outputCol), outputDataType, nullable = false)
StructType(outputFields) StructType(outputFields)
} }