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: [email protected]
For additional commands, e-mail: [email protected]