def METHOD_NAME(self):
    raw_dataset = load_dataset("squad", "plain_text", split='train', streaming=True)
    iterable_dataset = raw_dataset.take(100)
    examples = {}
    for e in iterable_dataset:
        for key in e.keys():
            if key not in examples:
                examples[key] = []
            examples[key].append(e[key])
    return Dataset.from_dict(examples)