src/transformers/pipelines/pt_utils.py/KeyPairDataset/__getitem__
class KeyPairDataset:    def __getitem__(self, i):
        return {"text": self.dataset[i][self.key1], "text_pair": self.dataset[i][self.key2]}