bugfix

avulanov · Nov 4, 2014 · 1900085 · 1900085
1 parent bb22442
commit 1900085
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/core/src/main/scala/org/apache/spark/api/python/PythonRDD.scala b/core/src/main/scala/org/apache/spark/api/python/PythonRDD.scala
@@ -405,15 +405,21 @@ private[spark] object PythonRDD extends Logging {
             dataOut.writeInt(bytes.length)
             dataOut.write(bytes)
           }
-        case pair: (String, PortableDataStream) =>
+        case (key: String, stream: PortableDataStream) =>
           newIter.asInstanceOf[Iterator[(String, PortableDataStream)]].foreach {
             case (key, stream) =>
               writeUTF(key, dataOut)
               val bytes = stream.toArray()
               dataOut.writeInt(bytes.length)
               dataOut.write(bytes)
           }
-        case pair: (Array[Byte], Array[Byte]) =>
+        case (key: String, value: String) =>
+          newIter.asInstanceOf[Iterator[(String, String)]].foreach {
+            case (key, value) =>
+              writeUTF(key, dataOut)
+              writeUTF(value, dataOut)
+          }
+        case (key: Array[Byte], value: Array[Byte]) =>
           newIter.asInstanceOf[Iterator[(Array[Byte], Array[Byte])]].foreach {
             case (key, value) =>
               dataOut.writeInt(key.length)