wip: implementing & using PartitionedBatchQueue
partition_by_key is expected to be work...