Skip to content

Commit 17ebe6b

Browse files
committed
Allow infinite buffer size for memory object streams in kernel
1 parent fb7b960 commit 17ebe6b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ipykernel/kernelbase.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import inspect
88
import itertools
99
import logging
10+
import math
1011
import os
1112
import queue
1213
import sys
@@ -424,7 +425,7 @@ async def shell_main(self, subshell_id: str | None):
424425
assert subshell_id is None
425426
assert threading.current_thread() == threading.main_thread()
426427
socket = None
427-
send_stream, receive_stream = create_memory_object_stream()
428+
send_stream, receive_stream = create_memory_object_stream(max_buffer_size=math.inf)
428429
async with create_task_group() as tg:
429430
if not socket.started.is_set():
430431
await tg.start(socket.start)

0 commit comments

Comments
 (0)