Skip to content

Commit 4c9c007

Browse files
committed
Revert changes to BatchEvalPythonExecSuite
1 parent 78f2ebf commit 4c9c007

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

sql/core/src/test/scala/org/apache/spark/sql/execution/python/BatchEvalPythonExecSuite.scala

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,19 +23,17 @@ import scala.collection.mutable.ArrayBuffer
2323
import org.apache.spark.api.python.{PythonEvalType, PythonFunction}
2424
import org.apache.spark.sql.catalyst.FunctionIdentifier
2525
import org.apache.spark.sql.catalyst.expressions.{And, AttributeReference, GreaterThan, In}
26-
import org.apache.spark.sql.execution._
26+
import org.apache.spark.sql.execution.{FilterExec, InputAdapter, SparkPlanTest, WholeStageCodegenExec}
2727
import org.apache.spark.sql.test.SharedSQLContext
2828
import org.apache.spark.sql.types.BooleanType
2929

3030
class BatchEvalPythonExecSuite extends SparkPlanTest with SharedSQLContext {
3131
import testImplicits.newProductEncoder
3232
import testImplicits.localSeqToDatasetHolder
3333

34-
val pythonUDF = new MyDummyPythonUDF
35-
3634
override def beforeAll(): Unit = {
3735
super.beforeAll()
38-
spark.udf.registerPython("dummyPythonUDF", pythonUDF)
36+
spark.udf.registerPython("dummyPythonUDF", new MyDummyPythonUDF)
3937
}
4038

4139
override def afterAll(): Unit = {

0 commit comments

Comments
 (0)