fixed missing input parameter training_data