def _handle_optimizer()

in scripts/old_config_converter.py [0:0]

• 15 lines of code
• 6 McCabe index (conditional complexity)

def _handle_optimizer(trainer):
    if "optimizer" not in trainer:
        return

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

    if "type" not in optimizer:
        pass
    elif "sgd" == optimizer["type"].lower():
        optimizer["_base_"] = "base_optimizer_sgd"
    elif "fedprox" == optimizer["type"].lower():
        optimizer["_base_"] = "base_optimizer_fedprox"
    optimizer.pop("type", None)