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__(