Great minds think alike - I will link those PRs.

Haha literally about 1.5 hours ago i sent the same pull request, but instead of wrapping the load_dataset() i renamed it, i liked yours better.

I also wasn’t sure whether it has been used somewhere else, so left the original intact.