import json def merge_predictions(results): if len(results) == 0: return {} formats = results[0][1].keys() predictions = {format_: {} for format_ in formats} for format_ in formats: for indices, batch_preds in results: for idx, preds in zip(indices, batch_preds[format_]): predictions[format_][idx] = preds predictions[format_] = [predictions[format_][i] for i in range(len(predictions[format_]))] return predictions