def METHOD_NAME(self, examples):
    dist_params, test_data = [], []
    for ex in examples:
        test_data.append(ex.pop("test_data"))
        dist_params.append(ex)
    return dist_params, test_data