diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/VectorAssembler.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/VectorAssembler.scala index 51acbf3806..3070012266 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/feature/VectorAssembler.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/feature/VectorAssembler.scala @@ -288,7 +288,7 @@ object VectorAssembler extends DefaultParamsReadable[VectorAssembler] { featureIndex += length } else { throw new SparkException( - s"""Encountered null while assembling a row with handleInvalid = "keep". Consider + s"""Encountered null while assembling a row with handleInvalid = "error". Consider |removing nulls from dataset or using handleInvalid = "keep" or "skip".""" .stripMargin) }