Trap timeouterror

This commit is contained in:
Jeremiah Lowin 2025-05-20 19:05:38 -04:00
commit f084f3e3db

View file

@ -1,4 +1,3 @@
import asyncio
import datetime
from contextlib import AsyncExitStack, asynccontextmanager
from pathlib import Path
@ -164,7 +163,7 @@ class Client:
with anyio.fail_after(1):
self._initialize_result = await self._session.initialize()
yield
except asyncio.TimeoutError:
except TimeoutError:
raise RuntimeError("Failed to initialize server session")
finally:
self._exit_stack = None