Github user viirya commented on a diff in the pull request: https://github.com/apache/spark/pull/19872#discussion_r154569177 --- Diff: sql/core/src/main/scala/org/apache/spark/sql/execution/python/ExtractPythonUDFs.scala --- @@ -113,6 +113,7 @@ object ExtractPythonUDFs extends Rule[SparkPlan] with PredicateHelper { def apply(plan: SparkPlan): SparkPlan = plan transformUp { // FlatMapGroupsInPandas can be evaluated directly in python worker // Therefore we don't need to extract the UDFs --- End diff -- `FlatMapGroupsInPandas` and `AggregateInPandasExec` can be...
--- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org