From ca39f26f27690c42ab567be2de7430590ee4813d Mon Sep 17 00:00:00 2001 From: drbh Date: Thu, 16 Jan 2025 18:22:02 +0000 Subject: [PATCH] fix: rerun pre commit lints --- integration-tests/models/test_flash_starcoder2_lora.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/integration-tests/models/test_flash_starcoder2_lora.py b/integration-tests/models/test_flash_starcoder2_lora.py index 878e7d24..6480f669 100644 --- a/integration-tests/models/test_flash_starcoder2_lora.py +++ b/integration-tests/models/test_flash_starcoder2_lora.py @@ -54,6 +54,7 @@ async def test_flash_starcoder2_load( assert responses == response_snapshot + @pytest.mark.asyncio async def test_flash_starcoder2_with_hugcode_adapter( flash_starcoder2, response_snapshot @@ -73,6 +74,6 @@ async def test_flash_starcoder2_with_hugcode_adapter( assert response.status_code == 200 data = response.json() - assert data["generated_text"] == "_world():\n print(\"Hello World!\")\n" + assert data["generated_text"] == '_world():\n print("Hello World!")\n' assert data == response_snapshot