Correct documentation examples of `optimizer_step` (#3326)
Correct documentation examples of `optimizer_step` (#3326)

* Correct documentation examples of `optimizer_step`

Without the default arguments set in `optimizer_step` the examples fail due to the arguments not being provided

* Apply suggestions from code review

Co-authored-by: Rohit Gupta <cb3c4569854e7764e66b4cfd98eab60a0e5fb067@gmail.com>

Co-authored-by: Nicki Skafte <6834dde87a397b36fec9709e4a16ddb6dba8b76b@gmail.com>
Co-authored-by: Rohit Gupta <cb3c4569854e7764e66b4cfd98eab60a0e5fb067@gmail.com>
williamFalcon/pytorch-lightning,williamFalcon/pytorch-lightning