Github user viirya commented on a diff in the pull request:
https://github.com/apache/spark/pull/21082#discussion_r183417848
--- Diff:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/PythonUDF.scala
---
@@ -34,10 +34,15 @@ object PythonUDF {
e.isInstanceOf[PythonUDF] &&
SCALAR_TYPES.contains(e.asInstanceOf[PythonUDF].evalType)
}
- def isGroupAggPandasUDF(e: Expression): Boolean = {
+ def isGroupedAggPandasUDF(e: Expression): Boolean = {
e.isInstanceOf[PythonUDF] &&
e.asInstanceOf[PythonUDF].evalType ==
PythonEvalType.SQL_GROUPED_AGG_PANDAS_UDF
}
+
+ def isWindowPandasUDF(e: Expression): Boolean = {
+ e.isInstanceOf[PythonUDF] &&
+ e.asInstanceOf[PythonUDF].evalType ==
PythonEvalType.SQL_GROUPED_AGG_PANDAS_UDF
--- End diff --
nit: indent style.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
