From f65ba6b4072b3e26375cc4d90f26b656e6c50099 Mon Sep 17 00:00:00 2001 From: Jason Liu Date: Wed, 7 Feb 2024 10:53:01 -0500 Subject: [PATCH] fix: test for multitask --- tests/openai/test_multitask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/openai/test_multitask.py b/tests/openai/test_multitask.py index 1700a4772..70e51abc1 100644 --- a/tests/openai/test_multitask.py +++ b/tests/openai/test_multitask.py @@ -71,7 +71,7 @@ async def stream_extract(input: str) -> Iterable[User]: ) resp = [] - async for user in await stream_extract(input="Jason is 20, Sarah is 30"): + for user in await stream_extract(input="Jason is 20, Sarah is 30"): resp.append(user) print(resp) assert len(resp) == 2