Skip to content

Commit c5a170d

Browse files
committed
change Array to JavaList when encountering Pickle
1 parent 993dc9c commit c5a170d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

mllib/src/main/scala/org/apache/spark/mllib/api/python/PythonMLLibAPI.scala

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1473,7 +1473,10 @@ private[spark] object SerDe extends Serializable {
14731473
initialize()
14741474

14751475
def dumps(obj: AnyRef): Array[Byte] = {
1476-
new Pickler().dumps(obj)
1476+
obj match {
1477+
case array: Array[_] => new Pickler().dumps(array.toSeq.asJava)
1478+
case _ => new Pickler().dumps(obj)
1479+
}
14771480
}
14781481

14791482
def loads(bytes: Array[Byte]): AnyRef = {

0 commit comments

Comments
 (0)