I've been keen to improve PartitionedDataSet in general so this could feed into a new design