Skip to content

Commit

Permalink
add type annotation
Browse files Browse the repository at this point in the history
  • Loading branch information
michaelfeil committed Feb 17, 2024
1 parent 6751ac7 commit a546f05
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ async def __aenter__(self) -> None:
"""
await self.engine.__aenter__()

async def __aexit__(self, *args) -> None:
async def __aexit__(self, *args: Any) -> None:
"""stop the background worker,
required to free references to the pytorch model."""
await self.engine.__aexit__(*args)
Expand Down

0 comments on commit a546f05

Please sign in to comment.