Skip to content

Commit

Permalink
Fixed aggregator.py
Browse files Browse the repository at this point in the history
Signed-off-by: Bruno Casella <[email protected]>
  • Loading branch information
CasellaJr committed Nov 14, 2023
1 parent ba705b3 commit a9a1266
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions openfl/component/aggregator/aggregator.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,8 +58,7 @@ def __init__(self,
**kwargs):
"""Initialize."""
#INITIALIZE WANDB WITH CORRECT NAME
#The following variable, my_aggregator_name, can be changed to whatever you want. Right now I suppose that the name of the collaborators of the federation is in the format
DATASETNAME_ENV_NUMBER
#The following variable, my_aggregator_name, can be changed to whatever you want. Right now I suppose that the name of the collaborators of the federation is in the format DATASETNAME_ENV_NUMBER
my_aggregator_name = '_'.join(set(element.split('_')[0] for element in authorized_cols))
if LOG_WANDB:
wandb.init(project="my_project", entity="my_group", group=f"{my_aggregator_name}", tags=["my_tag"],
Expand Down

0 comments on commit a9a1266

Please sign in to comment.