The only other instance of
on_training_epoch_end
I could find is in the paragraph above the code here in the docs
Thanks for checking! This got recently fixed already. You will see it if you replace “stable” with “latest” in the docs link you posted. So we should be good.