Major 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))
