Merge pull request FederatedAI#2740 from FederatedAI/feature-1.6-fix-…

…typo fix typo
jhdxr · Apr 16, 2021 · 656ff9c · 656ff9c
2 parents 178f04d + 66b0af3
commit 656ff9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/federatedml/ensemble/boosting/homo/homo_secureboost_client.py b/python/federatedml/ensemble/boosting/homo/homo_secureboost_client.py
@@ -156,7 +156,7 @@ def load_booster(self, model_meta, model_param, epoch_idx, booster_idx):
     def set_model_param(self, model_param):
         self.boosting_model_list = list(model_param.trees_)
         self.init_score = np.array(list(model_param.init_score))
-        self.classes_ = list(mpa(int, model_param.classes_))
+        self.classes_ = list(map(int, model_param.classes_))
         self.booster_dim = model_param.tree_dim
         self.num_classes = model_param.num_classes
         self.feature_name_fid_mapping.update(model_param.feature_name_fid_mapping)