The simplest way to iterate over a dataset is to just load one from 🤗 Datasets:

KeyDataset is a util that will just output the item we're interested in.