From ba33494aee07c9b9f20b4b367b74a21a9457001a Mon Sep 17 00:00:00 2001 From: Yuan Tang Date: Fri, 7 Feb 2025 23:18:24 -0500 Subject: [PATCH] edits Signed-off-by: Yuan Tang --- tests/client-sdk/inference/test_text_inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/client-sdk/inference/test_text_inference.py b/tests/client-sdk/inference/test_text_inference.py index d75686666..aa0e510dd 100644 --- a/tests/client-sdk/inference/test_text_inference.py +++ b/tests/client-sdk/inference/test_text_inference.py @@ -156,8 +156,8 @@ def test_text_completion_structured_output(llama_stack_client, text_model_id, in @pytest.mark.parametrize( "question,expected", [ - ("Which planet do humans live?", "Earth"), - ("Which planet has rings around them with a name starting with letter S?", "Saturn"), + ("Which planet do humans live on?", "Earth"), + ("Which planet has rings around it with a name starting with letter S?", "Saturn"), ], ) def test_text_chat_completion_non_streaming(llama_stack_client, text_model_id, question, expected):