Since we are not aliasing Dataset and IterableDataset anymore, this is possible and will make the code cleaner (especially the .pyi file).