message: wait for end aggregation message from the aggregator to enter finish state
diff: @@ -292,10 +292,6 @@ impl Protocol {
  if self.is_end_of_round() {
  self.emit_event(Event::RunAggregation);
  self.waiting_for_aggregation = true;
  - if self.current_round == self.settings.rounds {
  - info!("training complete");
  - self.is_training_complete = true;
  - } else {
  info!("round complete, resetting the clients");
  self.emit_event(Event::ResetAll);
  self.counters.waiting += self.counters.done;
  @@ -304,7 +300,6 @@ impl Protocol {
  self.counters.done = 0;
  self.counters.ignored = 0;
  }
  - }
  } else {
  self.emit_event(Event::SetState(id, ClientState::Ignored));
  self.counters.ignored += 1;
  @@ -322,6 +317,13 @@ impl Protocol {
  if success {
  self.current_round += 1;
  }
  + if self.current_round == self.settings.rounds {
  + info!("training complete");
  + self.is_training_complete = true;
  + } else {
  + info!("aggregation finished, proceeding to start a new round");
  + self.maybe_start_selection();
  + }
  }
  /// Retrieve the next event
  