def METHOD_NAME(self, local_agg_model_params, model_params, weight=1.0):
    # Add params that needed to be reduces from clients
    for name, param in model_params.items():
        if name not in local_agg_model_params:
            local_agg_model_params[name] = param * weight
        else:
            local_agg_model_params[name] += param * weight