From 48708247819a07809fe08895d42a0c9f84f9aea3 Mon Sep 17 00:00:00 2001 From: David Corvoysier Date: Mon, 4 Aug 2025 14:54:05 +0000 Subject: [PATCH] test(neuron): adjust expectations after graph changes --- integration-tests/neuron/test_generate.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/integration-tests/neuron/test_generate.py b/integration-tests/neuron/test_generate.py index d96aa36e..9d04309e 100644 --- a/integration-tests/neuron/test_generate.py +++ b/integration-tests/neuron/test_generate.py @@ -22,8 +22,8 @@ async def test_model_single_request(tgi_service): greedy_expectations = { "llama": " and how does it work?\nDeep learning is a subset of machine learning that uses artificial", "qwen2": " - Deep Learning is a subset of Machine Learning that involves the use of artificial neural networks", - "granite": "\n\nDeep learning is a subset of machine learning techniques based on artificial neural networks", - "qwen3": " A Deep Learning is a subset of machine learning that uses neural networks with multiple layers to", + "granite": "\n\nDeep Learning is a subset of machine learning that is inspired by the structure and", + "qwen3": " And Why Should You Care?\n\nDeep learning is a subset of machine learning that uses neural", "phi3": "\n\nDeep learning is a subfield of machine learning that focuses on creating", } assert response.generated_text == greedy_expectations[service_name] @@ -81,8 +81,8 @@ async def test_model_multiple_requests(tgi_service, neuron_generate_load): expectations = { "llama": "Deep learning is a subset of machine learning that uses artificial", "qwen2": "Deep Learning is a subset of Machine Learning that involves", - "granite": "Deep learning is a subset of machine learning techniques", - "qwen3": "Deep Learning is a subset of machine learning that uses neural networks", + "granite": "Deep Learning is a subset of machine learning that is inspired by the structure and", + "qwen3": " And Why Should You Care?\n\nDeep learning is a subset of machine learning that uses neural", "phi3": "Deep learning is a subfield of machine learning that focuses on creating", } expected = expectations[tgi_service.client.service_name]