Skip to content

Commit

Permalink
Don't assume len over 5 when all token ids are 0
Browse files Browse the repository at this point in the history
  • Loading branch information
g-eoj committed Jul 15, 2024
1 parent 6d1e428 commit a265a71
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/entrypoints/openai/test_completion.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ async def test_single_completion(client: openai.AsyncOpenAI, model_name: str,
max_tokens=5,
temperature=0.0,
)
assert len(completion.choices[0].text) >= 5
assert len(completion.choices[0].text) >= 1


@pytest.mark.asyncio
Expand Down

0 comments on commit a265a71

Please sign in to comment.