CDK-510 partitioned dataset

Should check to make sure dataset is a `PartitionedDataset` before casting.