requirements-cpu works

aidecentralized · Dec 2, 2024 · ab2fcc6 · ab2fcc6
1 parent d55dc12
commit ab2fcc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/algos/fl.py b/src/algos/fl.py
@@ -216,7 +216,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.