def METHOD_NAME(self, msg_params):
    logging.info("handle_message_receive_model_from_server.")
    model_params = msg_params.get(MyMessage.MSG_ARG_KEY_MODEL_PARAMS)
    client_index = msg_params.get(MyMessage.MSG_ARG_KEY_CLIENT_INDEX)
    self.trainer_dist_adapter.update_dataset(int(client_index))
    logging.info("current round index {}, total rounds {}".format(self.round_idx, self.num_rounds))
    self.trainer_dist_adapter.set_server_data(model_params)
    if self.round_idx < self.num_rounds:
        self.__train()
        self.round_idx += 1