Merge pull request #1094 from noopurintel/noopur/round-check

Logging fix - No of rounds saved vs started

.github/workflows/pytest_coverage.yml

@@ -1,11 +1,12 @@
-# This workflow will install Python dependencies, run tests and lint with a single version of Python
+# This workflow will run code coverage
 # For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
 
 name: Pytest and code coverage 
 
 on:
   pull_request:
     branches: [ develop ]
+  workflow_dispatch:
 
 permissions:
   contents: read

openfl/component/aggregator/aggregator.py

@@ -968,16 +968,17 @@ def _end_of_round_check(self):
 
         # Once all of the task results have been processed
         self._end_of_round_check_done[self.round_number] = True
+
+        # Save the latest model
+        self.logger.info("Saving round %s model...", self.round_number)
+        self._save_model(self.round_number, self.last_state_path)
+
         self.round_number += 1
         # resetting stragglers for task for a new round
         self.stragglers = []
         # resetting collaborators_done for next round
         self.collaborators_done = []
 
-        # Save the latest model
-        self.logger.info("Saving round %s model...", self.round_number)
-        self._save_model(self.round_number, self.last_state_path)
-
         # TODO This needs to be fixed!
         if self._time_to_quit():
             self.logger.info("Experiment Completed. Cleaning up...")