From 03836deacc2dc4327ef360324958d02d35c22ae3 Mon Sep 17 00:00:00 2001 From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com> Date: Tue, 16 May 2023 19:49:55 +0200 Subject: [PATCH] fix tests --- clients/python/tests/test_client.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/clients/python/tests/test_client.py b/clients/python/tests/test_client.py index 8972dfd1..32462f14 100644 --- a/clients/python/tests/test_client.py +++ b/clients/python/tests/test_client.py @@ -16,9 +16,9 @@ def test_generate(flan_t5_xxl_url, hf_headers): assert len(response.details.prefill) == 1 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 - ) + assert response.details.tokens[0].id == 3 + assert response.details.tokens[0].text == "" + assert not response.details.tokens[0].special def test_generate_best_of(flan_t5_xxl_url, hf_headers): @@ -82,9 +82,9 @@ async def test_generate_async(flan_t5_xxl_url, hf_headers): assert len(response.details.prefill) == 1 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 - ) + assert response.details.tokens[0].id == 3 + assert response.details.tokens[0].text == "" + assert not response.details.tokens[0].special @pytest.mark.asyncio