Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: replace expected output for continue test

This commit is contained in:
drbh 2024-11-25 10:10:33 -05:00
parent 594a6a7c22
commit 8505341931
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json
View File

@ -5,7 +5,7 @@
"index": 0,
"logprobs": null,
"message": {
"content": "the mouse is much smaller than an elephant. The average elephant weight is around 6,500 lbs (3,",
"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"
}
}

2
integration-tests/models/test_continue_final_message.py
View File

@ -71,6 +71,6 @@ def test_llama_completion_single_prompt_continue(
content = response["choices"][0]["message"]["content"]
assert (
content
== "the mouse is much smaller than an elephant. The average elephant weight is around 6,500 lbs (3,"
== " 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