write global weights for the initial round

xaynetwork · Feb 18, 2020 · f1efdc6 · f1efdc6
1 parent 5d52581
commit f1efdc6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/test_grpc.py b/tests/test_grpc.py
@@ -537,6 +537,7 @@ def mock_participant_store(*_args):
     )
     mock_local_part = mocker.patch("xain_sdk.participant.Participant")
     mock_local_part.init_weights.return_value = init_weight
+    mock_local_part.train_round.return_value = init_weight, 1
     mock_local_part.dummy_id = "participant1"
 
     config: Config = Config.from_unchecked_dict(participant_config)

diff --git a/xain_fl/coordinator/coordinator.py b/xain_fl/coordinator/coordinator.py
@@ -146,6 +146,9 @@ def __init__(  # pylint: disable=too-many-arguments
         self.current_round: int = 0
         self.epochs_current_round: int = epochs
 
+        # Write the weights for the initial round
+        self.global_weights_writer.write_weights(0, self.weights)
+
         self._write_metrics_fail_silently(
             {
                 "state": self.state,