@mariosasko Thank you for your help!
I found the solution to my stupid bug on “Tokens”. I transformed my sentences in “Tokens” into a list of tokens by adding the following line:
dataset = dataset.map(lambda ex: {“tokens”: ex[“tokens”].split(“,”)})
Thx again! : )