from typing import Callable


def get_lr_schedule_rnd(
    rnd: int, lr_initial: float, lr_decay: float
) -> Callable[[int], float]:
    """Return a schedule which decays the learning rate after each round."""
    lr_rnd = lr_initial * lr_decay ** rnd

    # pylint: disable-msg=unused-argument
    def lr_schedule(epoch: int) -> float:
        """Learning rate schedule."""
        return lr_rnd

    return lr_schedule