Labelling errors in the train set

Looking forward to using the amended version on WILDS! :grinning: