The comment doesn't need to say where the code was copy-pasted from, unless something about that is important for understanding the new implementation. I think all you really need to say here is that it's testing that Momentum gets the same values you get by hand, and that this example exercises the get_lr_scalers() functionality.