from torch.optim.lr_scheduler import (
    StepLR,
    CosineAnnealingLR,
    ReduceLROnPlateau,
    CyclicLR,
    CosineAnnealingWarmRestarts,
)
from src.utils.mapper import configmapper

configmapper.map("schedulers", "step")(StepLR)
configmapper.map("schedulers", "cosineanneal")(CosineAnnealingLR)
configmapper.map("schedulers", "reduceplateau")(ReduceLROnPlateau)
configmapper.map("schedulers", "cyclic")(CyclicLR)
configmapper.map("schedulers", "cosineannealrestart")(CosineAnnealingWarmRestarts)