Features should be updated when `map()` changes schema
`dataset.column_names` are being updated but `dataset.features` aren't indeed...
`dataset.map()` can change the schema and column names.

We should update the features in this case (with what is possible to infer).