include in gather wf the partition of the dataset