Correcting calls to get_sleep_time() (#1266)

securefederatedai · Jan 14, 2025 · fdad4fb · fdad4fb
1 parent 33004f6
commit fdad4fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py
@@ -472,7 +472,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
 
@@ -502,7 +502,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