Overhaul server architecture

learning-at-home · May 6, 2021 · 6fb6252 · 6fb6252
1 parent e25348c
commit 6fb6252
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/hivemind/server/runtime.py b/hivemind/server/runtime.py
@@ -118,8 +118,6 @@ def iterate_minibatches_from_pools(self, timeout=None):
                 logger.debug("Waiting for inputs from task pools")
                 ready_fds = selector.select()
                 ready_objects = {key.data for (key, events) in ready_fds}
-                if self.SHUTDOWN_TRIGGER in ready_objects:
-                    break  # someone asked us to shutdown, break from the loop
 
                 logger.debug("Choosing the pool with highest priority")
                 pool = max(ready_objects, key=lambda pool: pool.priority)

diff --git a/hivemind/server/task_pool.py b/hivemind/server/task_pool.py
@@ -67,7 +67,6 @@ class TaskPool(TaskPoolBase):
     :param timeout: wait for a subsequent task for at most this many seconds
     :param pool_size: store at most this many unprocessed tasks in a queue
     :param prefetch_batches: prepare up to this many *batches* in background for faster off-loading to runtime
-    :param uid: pool identifier used for shared array allocation
     :param start: if True, start automatically at the end of __init__
     """