def METHOD_NAME(self, model_full_dict: Dict[str, str]) -> None:
    updated_weights = {}
    for model, weight in self.model_to_weight.items():
        model_full_name = model_full_dict.get(model, model)
        updated_weights[model_full_name] = weight
    self.model_to_weight = updated_weights