Add pyUDT to MatrixUDT

Author: MechCoder

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

@@ -197,6 +197,8 @@ private[spark] class MatrixUDT extends UserDefinedType[Matrix] {
 
   override def typeName: String = "matrix"
 
+  override def pyUDT: String = "pyspark.mllib.linalg.MatrixUDT"
+
   private[spark] override def asNullable: MatrixUDT = this
 }
 

python/pyspark/mllib/linalg.py

@@ -199,7 +199,7 @@ def serialize(self, obj):
             return (1, obj.numRows, obj.numCols, None, None, values,
                     bool(obj.isTransposed))
         else:
-            raise TypeError("cannot serialize %r of type %r" % (obj, type(obj)))
+            raise TypeError("cannot serialize type %r" % (type(obj)))
 
     def deserialize(self, datum):
         assert len(datum) == 7, \

python/pyspark/mllib/tests.py

@@ -513,7 +513,6 @@ class MatrixUDTTests(MLlibTestCase):
     dm2 = DenseMatrix(3, 2, [0, 1, 4, 5, 9, 10], isTransposed=True)
     sm1 = SparseMatrix(1, 1, [0, 1], [0], [2.0])
     sm2 = SparseMatrix(2, 1, [0, 0, 1], [0], [5.0], isTransposed=True)
-
     udt = MatrixUDT()
 
     def test_json_schema(self):