From 70e41716bad37b4169baa49ee062d1b5efae0672 Mon Sep 17 00:00:00 2001 From: kta-intel Date: Fri, 15 Nov 2024 13:16:30 -0800 Subject: [PATCH] remove set_trace and fix spacing Signed-off-by: kta-intel --- .../workspace/plan/defaults/tasks_xgb.yaml | 14 +++++++------- openfl/component/aggregator/aggregator.py | 1 - 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/openfl-workspace/workspace/plan/defaults/tasks_xgb.yaml b/openfl-workspace/workspace/plan/defaults/tasks_xgb.yaml index b61942d4f7..7b14010eaa 100644 --- a/openfl-workspace/workspace/plan/defaults/tasks_xgb.yaml +++ b/openfl-workspace/workspace/plan/defaults/tasks_xgb.yaml @@ -1,21 +1,21 @@ aggregated_model_validation: function : validate_task - kwargs : - apply : global + kwargs : + apply : global metrics : - acc locally_tuned_model_validation: - function : validate_task - kwargs : - apply: local + function : validate_task + kwargs : + apply : local metrics : - acc train: function : train_task - kwargs : - metrics : + kwargs : + metrics : - loss aggregation_type : template : openfl.interface.aggregation_functions.FedBaggingXGBoost \ No newline at end of file diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py index b3be9f7f63..5e0d6a3dc0 100644 --- a/openfl/component/aggregator/aggregator.py +++ b/openfl/component/aggregator/aggregator.py @@ -863,7 +863,6 @@ def _prepare_trained(self, tensor_name, origin, round_number, report, agg_result new_model_report, ("model",), ) - # import pdb; pdb.set_trace() # Finally, cache the updated model tensor self.tensor_db.cache_tensor({final_model_tk: new_model_nparray})