diff --git a/mllib/src/main/scala/org/apache/spark/ml/attribute/attributes.scala b/mllib/src/main/scala/org/apache/spark/ml/attribute/attributes.scala index e479f169021d..32bf47764914 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/attribute/attributes.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/attribute/attributes.scala @@ -106,7 +106,7 @@ sealed abstract class Attribute extends Serializable { .withMetadata(existingMetadata) .putMetadata(AttributeKeys.ML_ATTR, withoutName.withoutIndex.toMetadataImpl()) .build() - StructField(name.get, DoubleType, nullable = false, newMetadata) + StructField(name.get, DoubleType, nullable = true, newMetadata) } /** Converts to a [[StructField]]. */