src/transformers/pipelines/pt_utils.py/KeyPairDataset/__init__
class KeyPairDataset:    def __init__(self, dataset: Dataset, key1: str, key2: str):
        self.dataset = dataset
        self.key1 = key1
        self.key2 = key2