transformers

edcc66d2 - Remove unnecessary columns for all dataset types in `Trainer` (#17166)

Remove unnecessary columns for all dataset types in `Trainer` (#17166) * Remove unneeded columns for IterableDataset * Add test * Update trainer tests * Edit docstring * Lint * Apply feedback * Apply feedback