Skip to content

GitLab

Unverified Commit fce52cef authored by Mario Šaško's avatar Mario Šaško Committed by GitHub
Browse files

Fix `remove_columns` in `text-classification` example (#29351)

parent 5080ab12
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/text-classification/run_classification.py
View file @ fce52cef
......@@ -422,7 +422,7 @@ def main():
for split in raw_datasets.keys():
for column in data_args.remove_columns.split(","):
logger.info(f"removing column {column} from split {split}")
raw_datasets[split].remove_columns(column)
raw_datasets[split] = raw_datasets[split].remove_columns(column)
if data_args.label_column_name is not None and data_args.label_column_name != "label":
for key in raw_datasets.keys():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment