Learning Rate Callback on Step Rather Than Epoch?

Question

I have been trying to write a custom learning rate scheduler that updates every step rather than every epoch.

I have managed to implement the following learning rate sche