def METHOD_NAME(self, msg_params):
    global_model_params = msg_params.get(MyMessage.MSG_ARG_KEY_MODEL_PARAMS)
    # client_index = msg_params.get(MyMessage.MSG_ARG_KEY_CLIENT_INDEX)
    average_weight_dict = msg_params.get(MyMessage.MSG_ARG_KEY_AVG_WEIGHTS)
    client_schedule = msg_params.get(MyMessage.MSG_ARG_KEY_CLIENT_SCHEDULE)
    client_indexes = client_schedule[self.worker_id]
    self.round_idx = 0
    self.__train(global_model_params, client_indexes, average_weight_dict)