def _handle_lr_scheduler_in_client()

in scripts/old_config_converter.py [0:0]

• 17 lines of code
• 7 McCabe index (conditional complexity)

def _handle_lr_scheduler_in_client(client):
    if "local_lr_scheduler" not in client:
        return

    client["lr_scheduler"] = client["local_lr_scheduler"]
    del client["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)