hmmm,  I thought about this but it feels like we should then only get that type from the index.

Should we do that as well? or pass `dataset_type=None` to be honest?