Skip to content

Commit

Permalink
fix: stream -> _stream
Browse files Browse the repository at this point in the history
  • Loading branch information
phi-friday committed Aug 6, 2023
1 parent 634972f commit 8f570ec
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions src/async_wrapper/queue.py
Original file line number Diff line number Diff line change
Expand Up @@ -94,23 +94,24 @@ def __init__(self, max_size: float | None = None) -> None:
def __init__(
self,
max_size: float | None = None,
stream: tuple[
*,
_stream: tuple[
MemoryObjectSendStream[ValueT],
MemoryObjectReceiveStream[ValueT],
]
| None = None,
) -> None:
if stream is None:
if _stream is None:
self._putter, self._getter = create_memory_object_stream(
max_buffer_size=max_size or math.inf,
)
else:
putter, getter = stream
putter, getter = _stream
if putter._closed or getter._closed: # noqa: SLF001
raise QueueBrokenError("putter or getter is closed")
if putter._state.buffer is not getter._state.buffer: # noqa: SLF001
raise QueueBrokenError("putter and getter has diff buffer.")
self._putter, self._getter = stream
self._putter, self._getter = _stream

self._close_putter: bool = True
self._close_getter: bool = True
Expand Down Expand Up @@ -313,7 +314,7 @@ def _clone(self, *, putter: bool, getter: bool) -> Queue[ValueT]:
raise RuntimeError("putter and getter are None.")
_putter = self._putter.clone() if putter else self._putter
_getter = self._getter.clone() if getter else self._getter
new = Queue(stream=(_putter, _getter)) # type: ignore
new = Queue(_stream=(_putter, _getter)) # type: ignore
new._close_putter = putter # noqa: SLF001
new._close_getter = getter # noqa: SLF001
return new
Expand Down

0 comments on commit 8f570ec

Please sign in to comment.