From 9b9614cea3de541e8e0e7b8898d5bd38cc1db483 Mon Sep 17 00:00:00 2001 From: drbh Date: Tue, 14 May 2024 03:16:01 +0000 Subject: [PATCH] fix: small test tweak --- integration-tests/models/test_flash_pali_gemma.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/integration-tests/models/test_flash_pali_gemma.py b/integration-tests/models/test_flash_pali_gemma.py index 94b7b89d..e1fad0d3 100644 --- a/integration-tests/models/test_flash_pali_gemma.py +++ b/integration-tests/models/test_flash_pali_gemma.py @@ -34,6 +34,4 @@ async def test_flash_pali_gemma(flash_pali_gemma, response_snapshot): inputs = f"Where is the cow standing?\n![]({cow})" response = await flash_pali_gemma.generate(inputs, max_new_tokens=20) - # TODO: update this! this is incorrect and just to show the current state of the test - assert response.generated_text == " - HDS" - # assert response.generated_text == "\nbeach" + assert response.generated_text == "\nbeach"