Dask - return a dask.dataframe on map_partition call

Question

I\'m wondering How can I return a dask Dataframe when I call a map_partitions instead of a pd.Dataframe in order to avoid memory issues.

Input Dataframe<