utils/data.py

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