from datasets import load_dataset dialogsum = load_dataset('knkarthick/dialogsum') dataset = list(dialogsum['train']) + list(dialogsum['validation']) + list(dialogsum['test'])