Drop unnecessary exit stack

frankie567 · Dec 8, 2024 · fedb69d · fedb69d
1 parent e144d2a
commit fedb69d
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/httpx_ws/transport.py b/httpx_ws/transport.py
@@ -43,11  43,8 @@ def __init__(self, app: ASGIApp, scope: Scope) -> None:
     async def __aenter__(
         self,
     ) -> tuple["ASGIWebSocketAsyncNetworkStream", bytes]:
-        self.exit_stack = contextlib.AsyncExitStack()
-        task_group = await self.exit_stack.enter_async_context(
-            anyio.create_task_group()
-        )
-        task_group.start_soon(self._run)
         self._task_group = await anyio.create_task_group().__aenter__()
         self._task_group.start_soon(self._run)
 
         await self.send({"type": "websocket.connect"})
         message = await self.receive()
@@ -61,7  58,7 @@ async def __aenter__(
 
     async def __aexit__(self, exc_type, exc_val, exc_tb) -> None:
         await self.aclose()
-        await self.exit_stack.__aexit__(exc_type, exc_val, exc_tb)
         await self._task_group.__aexit__(exc_type, exc_val, exc_tb)
 
     async def read(
         self, max_bytes: int, timeout: typing.Optional[float] = None