Skip to content

Commit

Permalink
Loosen expected type (#18)
Browse files Browse the repository at this point in the history
  • Loading branch information
jonathangreen authored Mar 8, 2024
1 parent 00043ca commit 89f9d52
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/palace_tools/utils/http/streaming.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from typing import Any, BinaryIO, ContextManager, TypeVar

import rich.progress
from httpx import Response
from httpx import AsyncClient, Response

from palace_tools.utils.http.async_client import HTTPXAsyncClient

Expand Down Expand Up @@ -47,7 +47,7 @@ async def streaming_fetch(
progress_updaters: Callable[[int], Any]
| Sequence[Callable[[int], Any]]
| None = None,
http_client: HTTPXAsyncClient | None = None,
http_client: AsyncClient | None = None,
raise_for_status: bool = False,
) -> Response:
async with HTTPXAsyncClient.with_existing_client(http_client) as client:
Expand Down Expand Up @@ -80,7 +80,7 @@ async def streaming_fetch_with_progress(
task_label: str | None = None,
total_setters: Callable[[int], Any] | list[Callable[[int], Any]] | None = None,
progress_updaters: Callable[[int], Any] | list[Callable[[int], Any]] | None = None,
http_client: HTTPXAsyncClient | None = None,
http_client: AsyncClient | None = None,
raise_for_status: bool = False,
) -> Response:
_progress_bar: ContextManager[rich.progress.Progress] | None = None
Expand Down

0 comments on commit 89f9d52

Please sign in to comment.