Skip to content

Commit

Permalink
bump mypy to 1.10 (#398)
Browse files Browse the repository at this point in the history
  • Loading branch information
pmeier committed Apr 29, 2024
1 parent e0fe014 commit f947f2d
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion environment-dev.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ dependencies:
- pytest >=6
- pytest-mock
- pytest-asyncio
- mypy ==1.6.1
- mypy ==1.10.0
- pre-commit
- types-aiofiles
- sqlalchemy-stubs
Expand Down
4 changes: 2 additions & 2 deletions ragna/assistants/_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,13 +32,13 @@ def __init__(self) -> None:
async def answer(
self, prompt: str, sources: list[Source], *, max_new_tokens: int = 256
) -> AsyncIterator[str]:
async for chunk in self._call_api( # type: ignore[attr-defined, misc]
async for chunk in self._call_api(
prompt, sources, max_new_tokens=max_new_tokens
):
yield chunk

@abc.abstractmethod
async def _call_api(
def _call_api(
self, prompt: str, sources: list[Source], *, max_new_tokens: int
) -> AsyncIterator[str]: ...

Expand Down

0 comments on commit f947f2d

Please sign in to comment.