fix test: rollback changes for pyspark

Author: Davies Liu

sql/core/src/main/scala/org/apache/spark/sql/DataFrame.scala

@@ -1501,8 +1501,8 @@ class DataFrame private[sql](
    */
   protected[sql] def javaToPython: JavaRDD[Array[Byte]] = {
     val fieldTypes = schema.fields.map(_.dataType)
-    val rdd = queryExecution.executedPlan.execute().map(EvaluatePython.rowToArray(_, fieldTypes))
-    SerDeUtil.javaToPython(rdd.toJavaRDD())
+    val jrdd = rdd.map(EvaluatePython.rowToArray(_, fieldTypes)).toJavaRDD()
+    SerDeUtil.javaToPython(jrdd)
   }
 
   ////////////////////////////////////////////////////////////////////////////

sql/core/src/main/scala/org/apache/spark/sql/execution/pythonUdfs.scala

@@ -157,9 +157,9 @@ object EvaluatePython {
   }
 
   /**
-   * Convert InternalRow into Java Array (for pickled into Python)
+   * Convert Row into Java Array (for pickled into Python)
    */
-  def rowToArray(row: InternalRow, fields: Seq[DataType]): Array[Any] = {
+  def rowToArray(row: Row, fields: Seq[DataType]): Array[Any] = {
     // TODO: this is slow!
     row.toSeq.zip(fields).map {case (obj, dt) => toJava(obj, dt)}.toArray
   }