Correcting calls to get_sleep_time()

securefederatedai · Jan 13, 2025 · 9b7556e · 9b7556e
1 parent dfe9512
commit 9b7556e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py
@@ -366,7 +366,7 @@ def get_tasks(self, collaborator_name):
         # if no tasks, tell the collaborator to sleep
         if len(tasks) == 0:
             tasks = None
-            sleep_time = self._get_sleep_time()
+            sleep_time = Aggregator._get_sleep_time()
 
             return tasks, self.round_number, sleep_time, time_to_quit
 
@@ -396,7 +396,7 @@ def get_tasks(self, collaborator_name):
         # been completed
         if len(tasks) == 0:
             tasks = None
-            sleep_time = self._get_sleep_time()
+            sleep_time = Aggregator._get_sleep_time()
 
             return tasks, self.round_number, sleep_time, time_to_quit