Re: Purpose of type in pandas_udf

2020-11-12 Thread Sean Owen
It's the return value On Thu, Nov 12, 2020 at 5:20 PM Daniel Stojanov wrote: > Hi, > > > Note "double" in the function decorator. Is this specifying the type of > the data that goes into pandas_mean, or the type returned by that function? > > > Regards, > > > > > @pandas_udf("double", PandasUDFT

Purpose of type in pandas_udf

2020-11-12 Thread Daniel Stojanov
Hi, Note "double" in the function decorator. Is this specifying the type of the data that goes into pandas_mean, or the type returned by that function? Regards, @pandas_udf("double", PandasUDFType.GROUPED_AGG) def pandas_mean(v):     return v.sum() --