fix(test_scheduler.py): fix test

This commit is contained in:
Krrish Dholakia 2024-06-01 11:30:26 -07:00
parent a0fb301b18
commit c6559753c7
2 changed files with 4 additions and 9 deletions

View file

@ -1410,7 +1410,7 @@ async def test_parallel_streaming_requests(sync_mode, model):
]
def sync_test_streaming():
response: litellm.CustomStreamWrapper = litellm.acompletion( # type: ignore
response: litellm.CustomStreamWrapper = litellm.completion( # type: ignore
model=model,
messages=messages,
stream=True,