Sfoglia il codice sorgente

do not crash Averager.step on GeneratorExit

justheuristic 3 anni fa
parent
commit
218310dbdd
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      hivemind/averaging/allreduce.py

+ 1 - 1
hivemind/averaging/allreduce.py

@@ -294,7 +294,7 @@ class AllReduceRunner(ServicerBase):
             logger.debug(f"{self} - notifying peers about {averaging_pb2.MessageCode.Name(code)}")
             for peer_id, mode in zip(self.ordered_peer_ids, self.modes):
                 if peer_id != self.peer_id and mode != AveragingMode.CLIENT:
-                    pending_tasks.add(asyncio.create_task(self._send_error_to_peer(peer_id, code)))
+                    pending_tasks.add(asyncio.ensure_future(self._send_error_to_peer(peer_id, code)))
 
         if not self._future.done():
             if cancel: