Changed key_format param to partition_key_format to be more explicit