Skip to content

Commit 474d132

Browse files
MechCodermengxr
authored andcommitted
[SPARK-6308] [MLlib] [Sql] Override TypeName in VectorUDT and MatrixUDT
Author: MechCoder <[email protected]> Closes #5118 from MechCoder/spark-6308 and squashes the following commits: 6c8ffab [MechCoder] Add test for simpleString b966242 [MechCoder] [SPARK-6308] [MLlib][Sql] VectorUDT is displayed as vecto in dtypes
1 parent 9f3273b commit 474d132

File tree

4 files changed

+8
-0
lines changed

4 files changed

+8
-0
lines changed

mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala

+2
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,8 @@ private[spark] class MatrixUDT extends UserDefinedType[Matrix] {
187187

188188
override def hashCode(): Int = 1994
189189

190+
override def typeName: String = "matrix"
191+
190192
private[spark] override def asNullable: MatrixUDT = this
191193
}
192194

mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala

+2
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,8 @@ private[spark] class VectorUDT extends UserDefinedType[Vector] {
185185

186186
override def hashCode: Int = 7919
187187

188+
override def typeName: String = "vector"
189+
188190
private[spark] override def asNullable: VectorUDT = this
189191
}
190192

mllib/src/test/scala/org/apache/spark/mllib/linalg/MatricesSuite.scala

+2
Original file line numberDiff line numberDiff line change
@@ -436,5 +436,7 @@ class MatricesSuite extends FunSuite {
436436
Seq(dm1, dm2, dm3, sm1, sm2, sm3).foreach {
437437
mat => assert(mat.toArray === mUDT.deserialize(mUDT.serialize(mat)).toArray)
438438
}
439+
assert(mUDT.typeName == "matrix")
440+
assert(mUDT.simpleString == "matrix")
439441
}
440442
}

mllib/src/test/scala/org/apache/spark/mllib/linalg/VectorsSuite.scala

+2
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,8 @@ class VectorsSuite extends FunSuite {
187187
for (v <- Seq(dv0, dv1, sv0, sv1)) {
188188
assert(v === udt.deserialize(udt.serialize(v)))
189189
}
190+
assert(udt.typeName == "vector")
191+
assert(udt.simpleString == "vector")
190192
}
191193

192194
test("fromBreeze") {

0 commit comments

Comments
 (0)