Use the correct asyncio API

pull/21904/head
Lidi Zheng 5 years ago
parent df7d02c887
commit efe29d6945
  1. 3
      src/python/grpcio_tests/tests_aio/benchmark/benchmark_client.py

@ -16,6 +16,7 @@
import abc
import asyncio
import time
import logging
import grpc
from grpc.experimental import aio
@ -131,7 +132,7 @@ class StreamingAsyncBenchmarkClient(BenchmarkClient):
self._running = True
senders = (
self._one_streamming_call() for _ in range(self._concurrency))
await asyncio.wait(senders)
await asyncio.gather(*senders)
self._stopped.set()
async def stop(self):

Loading…
Cancel
Save