def _handle_lr_scheduler()

in scripts/old_config_converter.py [0:0]

• 19 lines of code
• 8 McCabe index (conditional complexity)

def _handle_lr_scheduler(trainer):
    if "local_lr_scheduler" not in trainer:
        return

    trainer["client"] = trainer["client"] if "client" in trainer else {}
    client = trainer["client"]
    client["lr_scheduler"] = trainer["local_lr_scheduler"]
    del trainer["local_lr_scheduler"]
    lr_scheduler = client["lr_scheduler"]

    if "type" not in lr_scheduler:
        pass
    elif "constant" == lr_scheduler["type"].lower():
        lr_scheduler["_base_"] = "base_constant_lr_scheduler"
    elif "default" == lr_scheduler["type"].lower():
        lr_scheduler["_base_"] = "base_constant_lr_scheduler"
    elif "batch_size_normalizer" == lr_scheduler["type"].lower():
        lr_scheduler["_base_"] = "base_lr_batch_size_normalizer_scheduler"
    elif "armijo_line_search" == lr_scheduler["type"].lower():
        lr_scheduler["_base_"] = "base_armijo_line_search_lr_scheduer"
    lr_scheduler.pop("type", None)