From 0b8208084998be7c4053847a7bace149c829b9e7 Mon Sep 17 00:00:00 2001 From: drbh Date: Tue, 16 Apr 2024 14:52:39 +0000 Subject: [PATCH] fix: adjust assert typo --- integration-tests/models/test_completion_prompts.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/integration-tests/models/test_completion_prompts.py b/integration-tests/models/test_completion_prompts.py index feec79da..496bd062 100644 --- a/integration-tests/models/test_completion_prompts.py +++ b/integration-tests/models/test_completion_prompts.py @@ -39,7 +39,7 @@ def test_flash_llama_completion_single_prompt( response = response.json() assert len(response["choices"]) == 1 - return response == response_snapshot + assert response == response_snapshot def test_flash_llama_completion_many_prompts(flash_llama_completion, response_snapshot): @@ -61,7 +61,7 @@ def test_flash_llama_completion_many_prompts(flash_llama_completion, response_sn all_indexes.sort() assert all_indexes == [0, 1, 2, 3] - return response == response_snapshot + assert response == response_snapshot async def test_flash_llama_completion_many_prompts_stream( @@ -100,4 +100,4 @@ async def test_flash_llama_completion_many_prompts_stream( assert 0 <= c["choices"][0]["index"] <= 4 assert response.status == 200 - return response == response_snapshot + assert response == response_snapshot