fixed #90: when stopping the run, window crashed.

Author: alex.liang

python/fedml/cli/edge_deployment/client_login.py

@@ -299,14 +299,16 @@ def run(self):
 
     def reset_devices_status(self, edge_id):
         self.mlops_metrics.report_client_training_status(edge_id, MqttManager.MSG_MLOPS_CLIENT_STATUS_FINISHED)
-        time.sleep(3)
 
     def stop_run(self):
         self.setup_client_mqtt_mgr()
 
         self.reset_devices_status(self.edge_id)
 
-        FedMLClientRunner.cleanup_learning_process()
+        try:
+            FedMLClientRunner.cleanup_learning_process()
+        except Exception as e:
+            pass
         click.echo("Stop run successfully.")
 
     def setup_client_mqtt_mgr(self):
@@ -374,7 +376,10 @@ def callback_stop_train(self, topic, payload):
 
         # Stop cross-silo server with multi processing mode
         self.request_json = request_json
-        multiprocessing.Process(target=self.stop_run).start()
+        try:
+            multiprocessing.Process(target=self.stop_run).start()
+        except Exception as e:
+            pass
 
     def cleanup_client_with_finished_status(self):
         self.setup_client_mqtt_mgr()