Skip to content

Commit

Permalink
fix(federeated_learning paradigm): fix the pylint check error
Browse files Browse the repository at this point in the history
Signed-off-by: Marchons <[email protected]>
  • Loading branch information
Yoda-wu committed Sep 26, 2024
1 parent 692ad0c commit 13a8bd8
Showing 1 changed file with 6 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,10 @@ def init_client(self):
for i in range(self.clients_number)
]

# pylint: disable=C0103
# pylint: disable=C0206
# pylint: disable=C0201
# pylint: disable=W1203
def run(self):
"""
run the test flow of incremental learning paradigm.
Expand All @@ -90,8 +94,8 @@ def run(self):
dataset_files = self._split_dataset(1) # only one split ——all the data
train_dataset_file, _ = dataset_files[0]
train_datasets = self.train_data_partition(train_dataset_file)
for round in range(self.rounds):
self.train(train_datasets, round=round)
for r in range(self.rounds):
self.train(train_datasets, round=r)
global_weights = self.aggregator.aggregate(self.aggregate_clients)
self.send_weights_to_clients(global_weights)
self.aggregate_clients.clear()
Expand Down

0 comments on commit 13a8bd8

Please sign in to comment.