They are dicts by design, perhaps this is a situation where it would make sense to inherit and extend the

PartitionedDataSet

class for your own purposes?