CDK-238 deprecate partition key

This doesn't quite look right because it adds `set` to the public API. Is this needed?