def METHOD_NAME(self, idx, data, outputs):
    return [
        [
            data["participant_id"][idx],
            data["session_id"][idx],
            data[f"{self.mode}_id"][idx].item(),
            data["label"][idx].item(),
            outputs[idx].item(),
        ]
    ]