From 4069955e44aa75acb370e8c53cbc238109b1ac3e Mon Sep 17 00:00:00 2001 From: drbh Date: Thu, 21 Nov 2024 12:04:24 -0500 Subject: [PATCH] fix: bump test output --- .../test_llama_completion_single_prompt_continue.json | 2 +- integration-tests/models/test_continue_final_message.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json b/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json index f880dd74..8f782694 100644 --- a/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json +++ b/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json @@ -5,7 +5,7 @@ "index": 0, "logprobs": null, "message": { - "content": " Shere Kohan's fantastic exploits? written by David Shimomura & illustrated by Sarah Stern\n\nTitle: Elephant", + "content": " the royal mouse? It is a little more slender and only weighs around 1.5 pounds for males and 1.3 pounds", "role": "assistant" } } diff --git a/integration-tests/models/test_continue_final_message.py b/integration-tests/models/test_continue_final_message.py index 88d94433..2fb99273 100644 --- a/integration-tests/models/test_continue_final_message.py +++ b/integration-tests/models/test_continue_final_message.py @@ -71,6 +71,6 @@ def test_llama_completion_single_prompt_continue( content = response["choices"][0]["message"]["content"] assert ( content - == " Shere Kohan's fantastic exploits? written by David Shimomura & illustrated by Sarah Stern\n\nTitle: Elephant" + == " the royal mouse? It is a little more slender and only weighs around 1.5 pounds for males and 1.3 pounds" ) assert response == response_snapshot