Only check type name.

viirya · viirya · commit 0e07bb879d4d · 2015-02-24T08:49:45.000+08:00
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/util/modelSaveLoad.scala b/mllib/src/main/scala/org/apache/spark/mllib/util/modelSaveLoad.scala
@@ -110,7 +110,7 @@ private[mllib] object Loader {
       assert(loadedFields.contains(field.name), s"Unable to parse model data." +
         s"  Expected field with name ${field.name} was missing in loaded schema:" +
         s" ${loadedFields.mkString(", ")}")
-      assert(loadedFields(field.name) == field.dataType,
+      assert(loadedFields(field.name).typeName == field.dataType.typeName,
         s"Unable to parse model data.  Expected field $field but found field" +
           s" with different type: ${loadedFields(field.name)}")
     }

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ private[mllib] object Loader {`
`110`	`110`	`assert(loadedFields.contains(field.name), s"Unable to parse model data." +`
`111`	`111`	`s" Expected field with name ${field.name} was missing in loaded schema:" +`
`112`	`112`	`s" ${loadedFields.mkString(", ")}")`
`113`		`- assert(loadedFields(field.name) == field.dataType,`
	`113`	`+ assert(loadedFields(field.name).typeName == field.dataType.typeName,`
`114`	`114`	`s"Unable to parse model data. Expected field $field but found field" +`
`115`	`115`	`s" with different type: ${loadedFields(field.name)}")`
`116`	`116`	`}`