test(tgi): update docker test expectations

huggingface · Sep 17, 2024 · 7884f87 · 7884f87
1 parent b4161dc
commit 7884f87
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/text-generation-inference/tests/integration/test_generate.py b/text-generation-inference/tests/integration/test_generate.py
@@ -45,9 +45,9 @@ async def test_model_single_request(tgi_service):
         seed=42,
     )
     sample_expectations = {
-        "gpt2": "A lot of researchers have tried to make",
-        "llama": "Deep Learning is a subset of Artificial Intelligence",
-        "mistral": "Deep Learning is a kind of machine learning",
+        "gpt2": "Deep Learning",
+        "llama": "Deep Learning",
+        "mistral": "Deep Learning",
     }
     assert sample_expectations[service_name] in response