From 7695acdab4148e72bc6c39eb10f2de743f943870 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gabriel=20Mart=C3=ADn=20Bl=C3=A1zquez?=
 <gmartinbdev@gmail.com>
Date: Fri, 17 Jan 2025 08:27:42 +0100
Subject: [PATCH] Fix unit tests

---
 tests/unit/models/llms/test_vllm.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/unit/models/llms/test_vllm.py b/tests/unit/models/llms/test_vllm.py
index 2230186bf..f21c0a969 100644
--- a/tests/unit/models/llms/test_vllm.py
+++ b/tests/unit/models/llms/test_vllm.py
@@ -106,8 +106,8 @@ class TestvLLM:
     @pytest.mark.parametrize(
         "multi_structured_output",
         # TODO:  uncomment once with update our code to work with `outlines>0.1.0`
-        # (True, False),
-        (False,),
+        (True, False),
+        # (False,),
     )
     @pytest.mark.parametrize(
         "num_generations, expected_result",
@@ -183,6 +183,7 @@ def test_generate(
 
         mocked_requests_output = [
             mock.Mock(  # RequestOutput
+                prompt_logprobs=[],
                 outputs=[
                     mock.Mock(  # CompletionOutput
                         text="I'm fine thank you",