Use dataset.data_specs in iterator setup