* Rename VisionDataset to NonGeoDataset * Keep VisionDataset but add DeprecationWarning * mypy fix * More fixes * More fixes * cast types * Undo cast * Fix usage in test * No idea why... * Update more datasets