From f4872600f01ea9b4dba9fb4c95f7145f4dc419eb Mon Sep 17 00:00:00 2001 From: qbc Date: Wed, 1 Feb 2023 17:39:53 +0800 Subject: [PATCH] fix json issue (#506) --- .../vertical_fl/xgb_base/worker/evaluation_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/federatedscope/vertical_fl/xgb_base/worker/evaluation_wrapper.py b/federatedscope/vertical_fl/xgb_base/worker/evaluation_wrapper.py index cfce7a5db..330ba5ebf 100644 --- a/federatedscope/vertical_fl/xgb_base/worker/evaluation_wrapper.py +++ b/federatedscope/vertical_fl/xgb_base/worker/evaluation_wrapper.py @@ -185,7 +185,7 @@ def callback_func_for_feature_importance(self, message: Message): self.msg_buffer['eval'][state]['feature_importance'] = {} self.msg_buffer['eval'][state]['metrics'] = None self.msg_buffer['eval'][state]['feature_importance'].update( - {sender: feature_importance}) + {str(sender): feature_importance}) self._check_and_save_result() def callback_funcs_for_metrics(self, message: Message):