PDIO-424 Ability to add sanity check to training data classes