Skip to content

Commit

Permalink
chore(internal): fix lint issues in tests
Browse files Browse the repository at this point in the history
  • Loading branch information
RobertCraigie committed May 28, 2024
1 parent 00e34e7 commit d857640
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/lib/streaming/test_messages.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from __future__ import annotations

import os
from typing import TypeVar
from typing import Any, TypeVar, cast
from typing_extensions import Iterator, AsyncIterator, override

import httpx
Expand Down Expand Up @@ -125,7 +125,7 @@ def test_basic_response(self, respx_mock: MockRouter) -> None:
event_handler=SyncEventTracker,
) as stream:
with pytest.warns(DeprecationWarning):
assert isinstance(stream, Stream)
assert isinstance(cast(Any, stream), Stream)

assert_basic_response(stream, stream.get_final_message())

Expand Down Expand Up @@ -167,7 +167,7 @@ async def test_basic_response(self, respx_mock: MockRouter) -> None:
event_handler=AsyncEventTracker,
) as stream:
with pytest.warns(DeprecationWarning):
assert isinstance(stream, AsyncStream)
assert isinstance(cast(Any, stream), AsyncStream)

assert_basic_response(stream, await stream.get_final_message())

Expand Down

0 comments on commit d857640

Please sign in to comment.