Yes, the transformation logic is indeed the same. How exactly would an implementation using a modular pipeline work? I would expect that an earlier pipeline "unpacks" the

PartitionedDataset

dict and maps each key to an input of the modular pipeline, is that the right way to look at it?