In my case the perfect solution was to have a column named “label”. My data set had the column “target”, “label_name” and so on. Seems like the column names made the issue.
I did toxic_encoded = toxic_encoded.rename_column("target", "label")
and it started training.