Fix Trainer for Datasets that don't have dict items#17239

sgugger merged 1 commit intomainhuggingface/transformers:mainfrom

fix_new_remove_colshuggingface/transformers:fix_new_remove_colsCopy head branch name to clipboard