Forgot to add in AnnotationStatistics to TrainTestPipeline