瀏覽代碼

fix: await unary call

Denis Mazur 4 年之前
父節點
當前提交
56d0efd83d
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      hivemind/p2p/p2p_daemon.py

+ 1 - 1
hivemind/p2p/p2p_daemon.py

@@ -453,7 +453,7 @@ class P2P:
     ) -> Awaitable[TOutputProtobuf]:
 
         if not isinstance(input, AsyncIterableABC):
-            return self._call_unary_protobuf_handler(peer_id, name, input, output_protobuf_type)
+            return await self._call_unary_protobuf_handler(peer_id, name, input, output_protobuf_type)
 
         responses = self._iterate_protobuf_stream_handler(peer_id, name, input, output_protobuf_type)