diff --git a/src/bloomy/async_client.py b/src/bloomy/async_client.py index 61f77d4..c46bf5a 100644 --- a/src/bloomy/async_client.py +++ b/src/bloomy/async_client.py @@ -2,7 +2,7 @@ from __future__ import annotations -from typing import TYPE_CHECKING +from typing import TYPE_CHECKING, Self import httpx @@ -112,7 +112,7 @@ def __init__( self._client ) - async def __aenter__(self) -> AsyncClient: + async def __aenter__(self) -> Self: """Enter the async context manager. Returns: diff --git a/src/bloomy/client.py b/src/bloomy/client.py index b6d3d78..d3ff661 100644 --- a/src/bloomy/client.py +++ b/src/bloomy/client.py @@ -2,7 +2,7 @@ from __future__ import annotations -from typing import TYPE_CHECKING +from typing import TYPE_CHECKING, Self import httpx @@ -92,7 +92,7 @@ def __init__( self.issue = IssueOperations(self._client) self.headline = HeadlineOperations(self._client) - def __enter__(self) -> Client: + def __enter__(self) -> Self: """Context manager entry. Returns: