requirements-cpu works

aidecentralized · Dec 1, 2024 · 3ff5a40 · 3ff5a40
1 parent 706e427
commit 3ff5a40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/algos/fl.py b/src/algos/fl.py
@@ -219,7 +219,7 @@ def receive_and_aggregate(self):
         avg_wts = self.aggregate(reprs)
         self.set_representation(avg_wts)
 
-    def single_round(self, round: int, attack_start_round: int = 0, attack_end_round: int = 1, dump_file_name=""):
+    def single_round(self, round: int, attack_start_round: int = 0, attack_end_round: int = 1):
         """
         Runs the whole training procedure.