Skip to content

Commit 1e69f68

Browse files
committed
[CoreEngine] report the killed status when deleting the deployment.
1 parent bcf9881 commit 1e69f68

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def callback_delete_deployment(self, topic, payload):
111111

112112
# Get the launch job id
113113
ComputeCacheManager.get_instance().set_redis_params(self.redis_addr, self.redis_port, self.redis_password)
114-
launch_job_id = ComputeCacheManager.get_instance().get_gpu_cache().get_endpoint_run_id_map(self.run_id)
114+
launch_job_id = ComputeCacheManager.get_instance().get_gpu_cache().get_endpoint_run_id_map(model_msg_object.run_id)
115115

116116
# Delete SQLite records
117117
FedMLServerDataInterface.get_instance().delete_job_from_db(model_msg_object.run_id)
@@ -142,7 +142,9 @@ def callback_delete_deployment(self, topic, payload):
142142

143143
# Report the launch job status with killed status.
144144
if launch_job_id is not None:
145-
self.status_reporter.report_server_id_status(launch_job_id, GeneralConstants.MSG_MLOPS_SERVER_STATUS_KILLED)
145+
self.generate_status_report(model_msg_object.run_id, self.edge_id, server_agent_id=self.edge_id).\
146+
report_server_id_status(launch_job_id, GeneralConstants.MSG_MLOPS_SERVER_STATUS_KILLED,
147+
server_id=self.edge_id, server_agent_id=self.edge_id)
146148

147149
def callback_start_deployment(self, topic, payload):
148150
# noinspection PyBroadException

0 commit comments

Comments
 (0)