Skip to content

Commit

Permalink
try increasing queue size
Browse files Browse the repository at this point in the history
  • Loading branch information
dhirving committed Aug 29, 2024
1 parent 1d0c2f8 commit b45dec2
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ async def _stream_query_pages(
# exit_stack.
async with contextmanager_in_threadpool(exit_stack):
# `None` signals that there is no more data to send.
queue = asyncio.Queue[QueryExecuteResultData | None](1)
queue = asyncio.Queue[QueryExecuteResultData | None](2)
async with asyncio.TaskGroup() as tg:
tg.create_task(_enqueue_query_pages(ctx, spec, queue))
async for message in _dequeue_query_pages_with_keepalive(queue):
Expand Down

0 comments on commit b45dec2

Please sign in to comment.