|
|
@@ -279,7 +279,7 @@ class AllReduceRunner(ServicerBase):
|
|
|
error = averaging_pb2.AveragingData(group_id=self.group_id, code=code)
|
|
|
await afirst(await self._get_peer_stub(peer_id).rpc_aggregate_part(as_aiter(error)))
|
|
|
except Exception as e:
|
|
|
- logger.debug(f"Caught {e} when sending error {averaging_pb2.MessageCode.Name(code)} to {peer_id}.")
|
|
|
+ logger.debug(f"Caught {e} when sending error {averaging_pb2.MessageCode.Name(code)} to {peer_id}")
|
|
|
|
|
|
def finalize(self, *, cancel: bool = False, exception: Optional[BaseException] = None):
|
|
|
"""finish or terminate AllReduceRunner, propagate any errors / cancellations to peers."""
|
Aleksandr Borzunov