fixed #66: received three FINISH messages

FedML-AI · Jun 6, 2022 · 1980e0a · 1980e0a
1 parent d0588bc
commit 1980e0a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/python/fedml/cli/edge_deployment/client_login.py b/python/fedml/cli/edge_deployment/client_login.py
@@ -376,8 +376,6 @@ def callback_stop_train(self, topic, payload):
         self.request_json = request_json
         multiprocessing.Process(target=self.stop_run).start()
 
-        self.mlops_metrics.report_client_training_status(self.edge_id, MqttManager.MSG_MLOPS_CLIENT_STATUS_FINISHED)
-
     def cleanup_client_with_finished_status(self):
         self.setup_client_mqtt_mgr()
 
@@ -470,8 +468,6 @@ def callback_runner_id_status(self, topic, payload):
             self.request_json = request_json
             multiprocessing.Process(target=self.cleanup_client_with_finished_status).start()
 
-            self.mlops_metrics.report_client_training_status(self.edge_id, MqttManager.MSG_MLOPS_CLIENT_STATUS_FINISHED)
-
     @staticmethod
     def get_device_id():
         if "nt" in os.name:

diff --git a/python/fedml/cli/server_deployment/server_login.py b/python/fedml/cli/server_deployment/server_login.py
@@ -334,14 +334,10 @@ def reset_all_devices_status(self):
         edge_id_list = self.request_json["edgeids"]
         for edge_id in edge_id_list:
             self.mlops_metrics.report_client_training_status(edge_id, MqttManager.MSG_MLOPS_CLIENT_STATUS_IDLE)
-            time.sleep(1)
-        time.sleep(3)
 
     def stop_run(self):
         self.setup_client_mqtt_mgr()
 
-        self.reset_all_devices_status()
-
         FedMLServerRunner.cleanup_learning_process()
 
         try: