def _handle_async_weight()

in scripts/old_config_converter.py [0:0]

• 40 lines of code
• 16 McCabe index (conditional complexity)

def _handle_async_weight(trainer):  # noqa
    if (
        "staleness_weight_config" not in trainer
        and "example_weight_config" not in trainer
    ):
        return

    trainer["async_weight"] = {}
    async_weight = trainer["async_weight"]

    if "staleness_weight_config" in trainer:
        async_weight["staleness_weight"] = trainer["staleness_weight_config"]
        del trainer["staleness_weight_config"]
        staleness_weight = async_weight["staleness_weight"]

        if "type" not in staleness_weight:
            pass
        elif "default" == staleness_weight["type"].lower():
            staleness_weight["_base_"] = "base_constant_staleness_weight"
        elif "constant" == staleness_weight["type"].lower():
            staleness_weight["_base_"] = "base_constant_staleness_weight"
        elif "threshold" == staleness_weight["type"].lower():
            staleness_weight["_base_"] = "base_threshold_staleness_weight"
        elif "polynomial" == staleness_weight["type"].lower():
            staleness_weight["_base_"] = "base_polynomial_staleness_weight"
        staleness_weight.pop("type", None)

    if "example_weight_config" in trainer:
        async_weight["example_weight"] = trainer["example_weight_config"]
        del trainer["example_weight_config"]
        example_weight = async_weight["example_weight"]

        if "type" not in example_weight:
            pass
        elif "default" == example_weight["type"].lower():
            example_weight["_base_"] = "base_equal_example_weight"
        elif "equal" == example_weight["type"].lower():
            example_weight["_base_"] = "base_equal_example_weight"
        elif "linear" == example_weight["type"].lower():
            example_weight["_base_"] = "base_linear_example_weight"
        elif "sqrt" == example_weight["type"].lower():
            example_weight["_base_"] = "base_sqrt_example_weight"
        elif "log10" == example_weight["type"].lower():
            example_weight["_base_"] = "base_log10_example_weight"
        example_weight.pop("type", None)