I got burned by this too! Definitely seems like an issue in the library that .map doesn’t return the objects that the mapped function returns.
And I’d just finished the course which has a whole section about bugs caused by not setting return_tensors, but the examples never mention this issue with .map.