From 4616c629140cd7ff80ce6f29e8617def9c298471 Mon Sep 17 00:00:00 2001
From: fxmarty <9808326+fxmarty@users.noreply.github.com>
Date: Thu, 13 Jun 2024 10:57:31 +0000
Subject: [PATCH] style

---
 integration-tests/models/test_idefics2.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/integration-tests/models/test_idefics2.py b/integration-tests/models/test_idefics2.py
index b4f6d8cf..87fd5d30 100644
--- a/integration-tests/models/test_idefics2.py
+++ b/integration-tests/models/test_idefics2.py
@@ -3,6 +3,7 @@ import base64
 
 from testing_utils import require_backend_async, SYSTEM
 
+
 # TODO fix the server parsser to count inline image tokens correctly
 def get_chicken():
     with open("integration-tests/images/chicken_on_money.png", "rb") as image_file:
@@ -42,7 +43,7 @@ async def test_flash_idefics2_next_simple(flash_idefics2_next, response_snapshot
         response.generated_text == " A chicken is sitting on a pile of money."
     ), f"{repr(response.generated_text)}"
     assert response.details.generated_tokens == 10
-    
+
     if SYSTEM != "rocm":
         # Snapshot logprobs are not close enough on ROCm.
         assert response == response_snapshot
@@ -105,7 +106,7 @@ async def test_flash_idefics2_next_load(
     assert generated_texts[0] == " A chicken is sitting on a pile of money."
     assert len(generated_texts) == 4
     assert all([r.generated_text == generated_texts[0] for r in responses])
-    
+
     if SYSTEM != "rocm":
         # Snapshot logprobs are not close enough on ROCm.
         assert responses == response_snapshot