[CoreEngine] refactor the complete job callback.

Author: fedml-alex

python/fedml/computing/scheduler/master/base_master_protocol_manager.py

@@ -291,10 +291,10 @@ def callback_complete_job(self, topic, payload):
         if server_id is None:
             server_id = request_json.get("server_id", None)
 
-        # Complete the job runner
-        self._get_job_runner_manager().complete_job_runner(
-            run_id, args=self.args, server_id=server_id, request_json=request_json,
-            run_as_cloud_agent=self.run_as_cloud_agent, run_as_cloud_server=self.run_as_cloud_server)
+        self._process_job_complete_status(run_id, server_id, request_json)
+
+    def _process_job_complete_status(self, run_id, server_id, complete_payload):
+        pass
 
     def callback_run_logs(self, topic, payload):
         run_id = str(topic).split('/')[-1]

python/fedml/computing/scheduler/master/master_protocol_manager.py

@@ -34,3 +34,10 @@ def _init_extra_items(self):
     # Override
     def print_connected_info(self):
         super().print_connected_info()
+
+    # Override
+    def _process_job_complete_status(self, run_id, server_id, complete_payload):
+        # Complete the job runner
+        self._get_job_runner_manager().complete_job_runner(
+            run_id, args=self.args, server_id=server_id, request_json=complete_payload,
+            run_as_cloud_agent=self.run_as_cloud_agent, run_as_cloud_server=self.run_as_cloud_server)

python/fedml/computing/scheduler/model_scheduler/master_protocol_manager.py

@@ -103,9 +103,6 @@ def callback_deployment_result_message(self, topic=None, payload=None):
         logging.info(f"Received deployment result")
         FedMLDeployJobRunnerManager.get_instance().save_deployment_result(topic, payload)
 
-    def callback_complete_job(self, topic, payload):
-        pass
-
     def callback_delete_deployment(self, topic, payload):
         logging.info("[Master] callback_delete_deployment")
         # Parse payload as the model message object.