Bucketize shards based on bucket_key