Hello - I think as it stands this isn't possible out of the box. I need to check with the team why we made this a hard design requirement, but I would recommend you subclass PartitionedDataSet and override it for your own purposes.