Skip to content

Commit

Permalink
removed dump file name in fl
Browse files Browse the repository at this point in the history
  • Loading branch information
photonshi committed Dec 2, 2024
1 parent 6ef1cd4 commit 43a663a
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions src/algos/fl.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,12 +169,9 @@ def test(self, **kwargs: Any) -> Tuple[float, float, float]:
self.stats["test_loss"], self.stats["test_acc"], self.stats["test_time"] = test_loss, test_acc, time_taken
return test_loss, test_acc, time_taken

def receive_attack_and_aggregate(self, round: int, attack_start_round: int, attack_end_round: int, dump_file_name: str = ""):
def receive_attack_and_aggregate(self, round: int, attack_start_round: int, attack_end_round: int):
reprs = self.comm_utils.all_gather()

with open(dump_file_name, "wb") as f:
pickle.dump(reprs, f)

# Handle GIA-specific logic
if "gia" in self.config:
print("Server Running GIA attack")
Expand Down

0 comments on commit 43a663a

Please sign in to comment.