calling `self.log(..., on_step=True)` in optimizer_step fails
🐛 Bug
calling self.log(..., on_step=True) in optimizer_step fails.
Expected behavior
It should get treated same way if I were to .log() in training_step
Additional context
I believe it's connected to #4439. Tried debugging and hotfixing it but the whole .log() behavior (controlled by self._current_fx_name) seems not to be implemented too well.