Few test cases failing issue resolved

Signed-off-by: Parth Mandaliya <[email protected]>
securefederatedai · Jul 11, 2024 · 7b7726e · 7b7726e
1 parent 50ff1a2
commit 7b7726e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py
@@ -633,7 +633,8 @@ def send_local_task_results(self, collaborator_name, round_number, task_name,
             all_tasks_completed = (
                 all_tasks_completed and
                 self._collaborator_task_completed(
-                    collaborator=collaborator_name, task_name=task, round_num=self.round_number
+                    collaborator=collaborator_name, task_name=task.name,
+                    round_num=self.round_number
                 )
             )