roll-back since there is case of no model_id shows incorrect retraining count