Skip to content

Commit d1011a6

Browse files
authored
fixed the issue that the server can not stopped after training in the local development.
1 parent 3776c0e commit d1011a6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/fedml/cross_silo/client/fedml_client_master_manager.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,8 @@ def send_client_status(self, receive_id, status=ONLINE_STATUS_FLAG):
142142
message.add_params(MyMessage.MSG_ARG_KEY_CLIENT_STATUS, status)
143143
message.add_params(MyMessage.MSG_ARG_KEY_CLIENT_OS, sys_name)
144144

145-
if status == ClientMasterManager.RUN_FINISHED_STATUS_FLAG:
145+
if hasattr(self.args, "using_mlops") and self.args.using_mlops and \
146+
status == ClientMasterManager.RUN_FINISHED_STATUS_FLAG:
146147
mlops.log_server_payload(self.args.run_id, self.client_real_id, json.dumps(message.get_params()))
147148
else:
148149
self.send_message(message)

0 commit comments

Comments
 (0)