AttributeError: 'Dataset' object has no attribute 'remove_columns'

Beginners
1

Hi, I’m trying to remove a column of a huggingface dataset and I keep getting this message:
AttributeError: ‘Dataset’ object has no attribute ‘remove_columns’

The code:

from datasets import load_dataset

raw_datasets = load_dataset("glue", "mrpc")

raw_datasets['train'].remove_columns("sentence1")

I don’t get it, everywhere seems to point out this is the correct way but…

What am I missing?
Thank you guys

2

Hi ! It works fine on my side. What version of datasets are you using ?

3

Hi! I just solved it, it was indeed a problem of versions. I was using 1.0.1. With 1.14.0 it works fine.
Thanks!

4

You can use this line to solve it
df= df.remove_columns([“column_name”])