From 3462b5ea9533837007165f2309b35dcab10e912e Mon Sep 17 00:00:00 2001 From: mskarlin <12701035+mskarlin@users.noreply.github.com> Date: Fri, 14 Jun 2024 15:19:38 -0700 Subject: [PATCH] Update tests/test_paperqa.py Co-authored-by: James Braza --- tests/test_paperqa.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tests/test_paperqa.py b/tests/test_paperqa.py index 7c4bfa41d..766f41c39 100644 --- a/tests/test_paperqa.py +++ b/tests/test_paperqa.py @@ -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