Skip to content

Commit

Permalink
Fix for mypy error, closes #491
Browse files Browse the repository at this point in the history
This is why tests failed for #490
  • Loading branch information
simonw committed May 13, 2024
1 parent 3cc588f commit 0a8fd77
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion llm/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def __iter__(self) -> Iterator[str]:
self._start = time.monotonic()
self._start_utcnow = datetime.datetime.utcnow()
if self._done:
return self._chunks
yield from self._chunks
for chunk in self.model.execute(
self.prompt,
stream=self.stream,
Expand Down

0 comments on commit 0a8fd77

Please sign in to comment.