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 )