// Re-serialize the splits after grouping.
int count = 0;
for (InputSplit inputSplit : finalSplits) {
MRSplitProto serializedSplit = MRHelpers.createSplitProto(inputSplit);
RootInputDataInformationEvent diEvent = new RootInputDataInformationEvent(
count, serializedSplit.toByteArray());
diEvent.setTargetIndex(count);
count++;
taskEvents.add(diEvent);
}