update load_dataset for examples/feature_extraction

[sinchir0]

I fixed the error below.

Before

from datasets import load_dataset
load_dataset("smangrul/amazon_esci")

-> Error

datasets.exceptions.NonMatchingSplitsSizesError: [{'expected': SplitInfo(name='train', num_bytes=154877734, num_examples=839306, shard_lengths=None, dataset_name=None), 'recorded': SplitInfo(name='train', num_bytes=77438867, num_examples=419653, shard_lengths=None, dataset_name='amazon_esci')}, {'expected': SplitInfo(name='validation', num_bytes=66412102, num_examples=363402, shard_lengths=None, dataset_name=None), 'recorded': SplitInfo(name='validation', num_bytes=33206051, num_examples=181701, shard_lengths=None, dataset_name='amazon_esci')}]

After

from datasets import load_dataset
load_dataset("smangrul/amazon_esci", revision="main")

-> OK

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

[BenjaminBossan]

Thanks for fixing.