Method __init__

utils/data/data_utils.py:225–229

(self, prompt_dataset, answer_dataset)

class PromptDataset(Dataset):

    def __init__(self, prompt_dataset, answer_dataset) -> None:
        super().__init__()
        self.prompt_dataset = prompt_dataset
        self.answer_dataset = answer_dataset
        assert len(self.prompt_dataset) == len(self.answer_dataset)

    def __len__(self):
        return len(self.prompt_dataset)