Github user BryanCutler commented on a diff in the pull request:
https://github.com/apache/spark/pull/21650#discussion_r202864609
--- Diff:
sql/core/src/test/scala/org/apache/spark/sql/execution/python/BatchEvalPythonExecSuite.scala
---
@@ -23,21 +23,27 @@ import scala.collection.mutable.ArrayBuffer
import org.apache.spark.api.python.{PythonEvalType, PythonFunction}
import org.apache.spark.sql.catalyst.FunctionIdentifier
import org.apache.spark.sql.catalyst.expressions.{And, AttributeReference,
GreaterThan, In}
-import org.apache.spark.sql.execution.{FilterExec, InputAdapter,
SparkPlanTest, WholeStageCodegenExec}
+import org.apache.spark.sql.execution._
+import org.apache.spark.sql.functions.col
import org.apache.spark.sql.test.SharedSQLContext
import org.apache.spark.sql.types.BooleanType
class BatchEvalPythonExecSuite extends SparkPlanTest with SharedSQLContext
{
import testImplicits.newProductEncoder
import testImplicits.localSeqToDatasetHolder
+ val pythonUDF = new MyDummyPythonUDF
--- End diff --
`pythonUDF` -> `pythonBatchedUDF`
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
