Github user BryanCutler commented on the issue:
https://github.com/apache/spark/pull/22305
I think there is a typo in your example in the description
```
@pandas_udf('double', PandasUDFType.GROUPED_AGG)
def avg(v):
return v.mean()
return avg
```
I think you didn't mean to have `return avg`?
Also, I think exploring returning numpy arrays would be good, but lets
discuss elsewhere and I would remove that from your description if it isn't
supported as part of this.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
