We requested this feature in dask to allow our users to have column projection through our custom functions like filters. We should update our code to use this new feature.
Places that we could use it:
There could be other places that we use map_partitions too.