rollback changes in mllib

apache · Apr 19, 2015 · d17d634 · d17d634
1 parent 2e7566d
commit d17d634
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pyspark/mllib/common.py b/python/pyspark/mllib/common.py
@@ -23,7 +23,7 @@
 import py4j.protocol
 from py4j.protocol import Py4JJavaError
 from py4j.java_gateway import JavaObject
-from py4j.java_collections import JavaArray, JavaList
+from py4j.java_collections import ListConverter, JavaArray, JavaList
 
 from pyspark import RDD, SparkContext
 from pyspark.serializers import PickleSerializer, AutoBatchedSerializer
@@ -76,7 +76,7 @@ def _py2java(sc, obj):
     elif isinstance(obj, SparkContext):
         obj = obj._jsc
     elif isinstance(obj, list):
-        obj = [_py2java(sc, x) for x in obj]
+        obj = ListConverter().convert([_py2java(sc, x) for x in obj], sc._gateway._gateway_client)
     elif isinstance(obj, JavaObject):
         pass
     elif isinstance(obj, (int, long, float, bool, bytes, unicode)):