From c7803c41e8f51655a4bddd5544251ce12dbb5296 Mon Sep 17 00:00:00 2001 From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com> Date: Thu, 13 Apr 2023 16:52:12 +0200 Subject: [PATCH] fix test --- clients/python/tests/test_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clients/python/tests/test_client.py b/clients/python/tests/test_client.py index c998de41..d53c2a4d 100644 --- a/clients/python/tests/test_client.py +++ b/clients/python/tests/test_client.py @@ -17,7 +17,7 @@ def test_generate(flan_t5_xxl_url, hf_headers): assert response.details.prefill[0] == PrefillToken(id=0, text="", logprob=None) assert len(response.details.tokens) == 1 assert response.details.tokens[0] == Token( - id=3, text=" ", logprob=-1.984375, special=False + id=3, text="", logprob=-1.984375, special=False ) @@ -83,7 +83,7 @@ async def test_generate_async(flan_t5_xxl_url, hf_headers): assert response.details.prefill[0] == PrefillToken(id=0, text="", logprob=None) assert len(response.details.tokens) == 1 assert response.details.tokens[0] == Token( - id=3, text=" ", logprob=-1.984375, special=False + id=3, text="", logprob=-1.984375, special=False )