Fix unit tests

argilla-io · Jan 17, 2025 · 7695acd · 7695acd
1 parent 5ab734c
commit 7695acd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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",