From 9080f6885ed7bda3b025f7db157a6b4b290eec67 Mon Sep 17 00:00:00 2001
From: Alvaro Moran <alvaro.moran@huggingface.co>
Date: Wed, 28 Feb 2024 13:54:12 +0000
Subject: [PATCH] test fix WIP

---
 text-generation-inference/tests/test_generator.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/text-generation-inference/tests/test_generator.py b/text-generation-inference/tests/test_generator.py
index 085953cd..03cb377c 100644
--- a/text-generation-inference/tests/test_generator.py
+++ b/text-generation-inference/tests/test_generator.py
@@ -75,8 +75,8 @@ def create_request(
         ],
         [
             "It was a bright cold day in April, and the clocks were striking thirteen.",
-            1439,
-            " All",
+            775,
+            " We",
             True,
         ],
     ],
@@ -140,6 +140,7 @@ def test_decode_single(input_text, max_new_tokens, generated_text, do_sample, mo
     output = generations[0].generated_text
     assert output.generated_tokens == max_new_tokens
     assert output.finish_reason == 0
+    print(output.text)
     assert output.text == generated_text