From 8f570ec4b28ab916c4d8518bd57985cf2c23471a Mon Sep 17 00:00:00 2001 From: phi Date: Mon, 7 Aug 2023 00:25:13 +0900 Subject: [PATCH] fix: stream -> _stream --- src/async_wrapper/queue.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/async_wrapper/queue.py b/src/async_wrapper/queue.py index 7bb2332..31d76ca 100644 --- a/src/async_wrapper/queue.py +++ b/src/async_wrapper/queue.py @@ -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 @@ -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