Move PartitionedDataset to the SPI.