MT5 Fine Tuning - KeyError: 'source_ids'

Hi. I got this error when using Datasets library. Just changed it as nlp, then code worked. So,

instead of

from datasets import Dataset

I used

from nlp import Dataset

You can find the related script in here