Skip to content

Commit

Permalink
Update tests/test_paperqa.py
Browse files Browse the repository at this point in the history
Co-authored-by: James Braza <[email protected]>
  • Loading branch information
mskarlin and jamesbraza authored Jun 14, 2024
1 parent bba3aa0 commit 3462b5e
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions tests/test_paperqa.py
Original file line number Diff line number Diff line change
Expand Up @@ -580,11 +580,7 @@ async def test_anyscale_chain():
skip_system=True,
)
outputs = []

def accum(x):
outputs.append(x)

completion = await call({"animal": "duck"}, callbacks=[accum]) # type: ignore[call-arg]
completion = await call({"animal": "duck"}, callbacks=[outputs.append]) # type: ignore[call-arg]
assert completion.seconds_to_first_token > 0
assert completion.prompt_count > 0
assert completion.completion_count > 0
Expand Down

0 comments on commit 3462b5e

Please sign in to comment.