From debf032ca3b7563e0e46cb6bc966b4cbd8e87f6b Mon Sep 17 00:00:00 2001 From: David Corvoysier Date: Thu, 20 Feb 2025 17:26:56 +0000 Subject: [PATCH] test(neuron): no error anymore when requesting too many tokens --- .../neuron/integration/test_implicit_env.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/integration-tests/neuron/integration/test_implicit_env.py b/integration-tests/neuron/integration/test_implicit_env.py index df29b910..cf0c8062 100644 --- a/integration-tests/neuron/integration/test_implicit_env.py +++ b/integration-tests/neuron/integration/test_implicit_env.py @@ -53,17 +53,6 @@ async def test_model_single_request(tgi_service): ) assert response.details.generated_tokens == 17 - # check error - try: - await tgi_service.client.text_generation("What is Deep Learning?", max_new_tokens=170000) - except ValidationError: - pass - else: - raise AssertionError( - "The previous text generation request should have failed, " - "because too many tokens were requested, it succeeded" - ) - # Sampling await tgi_service.client.text_generation( "What is Deep Learning?",