Description
This function doesn't implement partial aggregation and shuffles all data. A uadf that supports partial aggregation is not covered by this Jira.
Exmaple:
@pandas_udf(DoubleType()) def mean(v) return v.mean() df.groupby('id').apply(mean(df.v1), mean(df.v2))