From 374d4efa8363717f3085c710fe6f767614caabaf Mon Sep 17 00:00:00 2001 From: Robert Steiner Date: Wed, 12 Feb 2020 13:54:13 +0100 Subject: [PATCH] PB-402 Add helper func --- xain_fl/coordinator/coordinator.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/xain_fl/coordinator/coordinator.py b/xain_fl/coordinator/coordinator.py index dfa6244c4..a62e1a186 100644 --- a/xain_fl/coordinator/coordinator.py +++ b/xain_fl/coordinator/coordinator.py @@ -277,8 +277,7 @@ def _handle_rendezvous( ) logger.debug( - "Send RendezvousResponse", - reply=_RENDEZVOUSREPLY.values_by_number[reply].name, + "Send RendezvousResponse", reply=pb_enum_to_str(_RENDEZVOUSREPLY, reply) ) return RendezvousResponse(reply=reply) @@ -311,7 +310,7 @@ def _handle_heartbeat( logger.debug( "Heartbeat response", participant_id=participant_id, - state=_STATE.values_by_number[state].name, + state=pb_enum_to_str(_STATE, state), round=self.current_round, current_participants_count=self.participants.len(), ) @@ -417,3 +416,16 @@ def _handle_end_training_round( logger.debug("Send EndTrainingRoundResponse", participant_id=participant_id) return EndTrainingRoundResponse() + + +def pb_enum_to_str(pb_enum, member_value: int) -> str: + """Return the human readable string of a enum member value. + + Args: + pb_enum: The proto enum definition. + member_value: The enum member value. + + Returns: + The human readable string of a enum member value. + """ + return pb_enum.values_by_number[member_value].name