Skip to content

Commit

Permalink
remove set_trace and fix spacing
Browse files Browse the repository at this point in the history
Signed-off-by: kta-intel <[email protected]>
  • Loading branch information
kta-intel committed Nov 15, 2024
1 parent ce4b34f commit 70e4171
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 8 deletions.
14 changes: 7 additions & 7 deletions openfl-workspace/workspace/plan/defaults/tasks_xgb.yaml
Original file line number Diff line number Diff line change
@@ -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
1 change: 0 additions & 1 deletion openfl/component/aggregator/aggregator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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})
Expand Down

0 comments on commit 70e4171

Please sign in to comment.