Describe the bug
The end_lr parameter is annotated as an int. However, it is used for two underlying schedulers that inherit from _LRSchedulerMonai, which uses a float.

Expected behavior
end_lr should be annotated as a float.