update checkInputColumn to print more info if needed

maropu · Jul 3, 2015 · 8c42b57 · 8c42b57
1 parent 33ddd2e
commit 8c42b57
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/util/SchemaUtils.scala b/mllib/src/main/scala/org/apache/spark/ml/util/SchemaUtils.scala
@@ -34,10 +34,11 @@ object SchemaUtils {
    * @param colName  column name
    * @param dataType  required column data type
    */
-  def checkColumnType(schema: StructType,
-                      colName: String,
-                      dataType: DataType,
-                      msg: String = ""): Unit = {
+  def checkColumnType(
+      schema: StructType,
+      colName: String,
+      dataType: DataType,
+      msg: String = ""): Unit = {
     val actualDataType = schema(colName).dataType
     val message = if (msg != null && msg.trim.length > 0) " " + msg else ""
     require(actualDataType.equals(dataType),