Fix model building input src -> training_frame