fix json issue (#506)

alibaba · Feb 1, 2023 · f487260 · f487260
1 parent 86d7dc3
commit f487260
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):