Remove unnecessary TODO comment in fit_loop.py

The TODO suggested moving max_steps check inside training loop, but this is
correctly placed in fit_loop since global_step tracks steps across epochs
and the fit_loop is responsible for deciding when to stop the entire fit
process. Moving it would require additional flag variables, making the code
more complex without any benefit.

Co-authored-by: Borda <[email protected]>