huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-19 22:02:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: update expected output from town to village

Browse Source
This commit is contained in:
drbh 2025-02-11 11:11:09 +01:00
parent 571ac9b507
commit 38b06b90b9
3 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
integration-tests/models/__snapshots__/test_mllama/test_mllama_load.json
View File

@ -6,7 +6,7 @@
"index": 0, "index": 0,
"logprobs": null, "logprobs": null,
"message": { "message": {
"content": "In a small town, a chicken named Cluck", "content": "In a small village, a chicken named Cluck",
"name": null, "name": null,
"role": "assistant", "role": "assistant",
"tool_calls": null "tool_calls": null
@ -32,7 +32,7 @@
"index": 0, "index": 0,
"logprobs": null, "logprobs": null,
"message": { "message": {
"content": "In a small town, a chicken named Cluck", "content": "In a small village, a chicken named Cluck",
"name": null, "name": null,
"role": "assistant", "role": "assistant",
"tool_calls": null "tool_calls": null

2
integration-tests/models/__snapshots__/test_mllama/test_mllama_simpl.json
View File

@ -5,7 +5,7 @@
"index": 0, "index": 0,
"logprobs": null, "logprobs": null,
"message": { "message": {
"content": "In a small town, a chicken named Cluck", "content": "In a small village, a chicken named Cluck",
"name": null, "name": null,
"role": "assistant", "role": "assistant",
"tool_calls": null "tool_calls": null

5
integration-tests/models/test_mllama.py
View File

@ -47,7 +47,8 @@ async def test_mllama_simpl(mllama, response_snapshot):
"total_tokens": 60, "total_tokens": 60,
} }
assert ( assert (
response.choices[0].message.content == "In a small town, a chicken named Cluck" response.choices[0].message.content
== "In a small village, a chicken named Cluck"
) )
assert response == response_snapshot assert response == response_snapshot
@ -86,7 +87,7 @@ async def test_mllama_load(mllama, generate_load, response_snapshot):
generated_texts = [response.choices[0].message.content for response in responses] generated_texts = [response.choices[0].message.content for response in responses]
# XXX: TODO: Fix this test. # XXX: TODO: Fix this test.
assert generated_texts[0] == "In a small town, a chicken named Cluck" assert generated_texts[0] == "In a small village, a chicken named Cluck"
assert len(generated_texts) == 2 assert len(generated_texts) == 2
assert generated_texts, all( assert generated_texts, all(
[text == generated_texts[0] for text in generated_texts] [text == generated_texts[0] for text in generated_texts]