style: add training
fix pending test