From b25936d4c5216904f0ca3cf33df4b5c7130aa8f8 Mon Sep 17 00:00:00 2001 From: Li Jin Date: Fri, 27 Jul 2018 13:58:06 +0000 Subject: [PATCH] minor fix --- .../apache/spark/sql/execution/python/ExtractPythonUDFs.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/python/ExtractPythonUDFs.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/python/ExtractPythonUDFs.scala index 38f2b31196e70..30fe71825db76 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/execution/python/ExtractPythonUDFs.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/execution/python/ExtractPythonUDFs.scala @@ -95,7 +95,7 @@ object ExtractPythonUDFFromAggregate extends Rule[LogicalPlan] { */ object ExtractPythonUDFs extends Rule[SparkPlan] with PredicateHelper { - private case class EvalTypeHolder(var evalType: Int = -1) { + private case class EvalTypeHolder(private var evalType: Int = -1) { def isSet: Boolean = evalType >= 0 @@ -135,7 +135,7 @@ object ExtractPythonUDFs extends Rule[SparkPlan] with PredicateHelper { case udf: PythonUDF if PythonUDF.isScalarPythonUDF(udf) && (!firstEvalType.isSet || firstEvalType.get == udf.evalType) && canEvaluateInPython(udf) => - firstEvalType.evalType = udf.evalType + firstEvalType.set(udf.evalType) Seq(udf) case e => e.children.flatMap(collectEvaluableUDFs(_, firstEvalType)) }