Caching a dataset with map() when loaded with from_dict()

Also stumbled on this bug. Any plans to fix it?