fix(test_scheduler.py): fix test

This commit is contained in:
Krrish Dholakia 2024-06-01 11:30:26 -07:00
parent de62c5f565
commit ca82b82d4b
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,