diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py index df0720ab03..d39828a5cc 100644 --- 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 ) )