Format

Signed-off-by: mgoin <[email protected]>
vllm-project · Dec 5, 2024 · 2966c19 · 2966c19
1 parent d00183a
commit 2966c19
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/entrypoints/llm/test_guided_generate.py b/tests/entrypoints/llm/test_guided_generate.py
@@ -106,10 +106,10 @@ def test_guided_complex_json_completion(sample_complex_json_schema, llm):
 
 @pytest.mark.skip_global_cleanup
 def test_guided_definition_json_completion(sample_definition_json_schema, llm):
-    sampling_params = SamplingParams(
-        temperature=1.0,
-        max_tokens=1000,
-        guided_decoding=GuidedDecodingParams(json=sample_definition_json_schema))
+    sampling_params = SamplingParams(temperature=1.0,
+                                     max_tokens=1000,
+                                     guided_decoding=GuidedDecodingParams(
+                                         json=sample_definition_json_schema))
     outputs = llm.generate(prompts=[
         f"Give an example JSON for solving 8x + 7 = -23 "
         f"that fits this schema: {sample_definition_json_schema}"