From b162e6908cd9ef2b4974eff06de4c4a853e5f2aa Mon Sep 17 00:00:00 2001 From: Nathaniel Hudson Date: Thu, 18 Jul 2024 22:09:29 -0500 Subject: [PATCH] Update fedavg.py --- flight/strategies/impl/fedavg.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/flight/strategies/impl/fedavg.py b/flight/strategies/impl/fedavg.py index 3bd3b2b..037ec27 100644 --- a/flight/strategies/impl/fedavg.py +++ b/flight/strategies/impl/fedavg.py @@ -48,11 +48,7 @@ def aggregate_params( class FedAvgWorker(DefaultWorkerStrategy): - """The worker for 'FedAvg' and its respective methods. - - Args: - DefaultWorkerStrategy: The base class providing necessary methods for 'FedAvgWorker' - """ + """The worker for 'FedAvg' and its respective methods.""" def before_training( self, state: NodeState, data: Params @@ -71,11 +67,9 @@ def before_training( class FedAvg(Strategy): - """Implementation of the FedAvg strategy, which uses default strategies for the trainer, - 'FedAvg' for aggregator and workers, and 'FedSGD' for the coordinator. - - Args: - Strategy: The base class providing the necessary attributes for 'FedAvg'. + """ + Implementation of the FedAvg strategy, which uses default strategies for the trainer, + 'FedAvg' for aggregator and workers, and 'FedSGD' for the coordinator. """ def __init__(