ENH: Add check_convergence flag to turn off warning