From 2966c192b5a74390754183e61fdef7dc500cac79 Mon Sep 17 00:00:00 2001 From: mgoin Date: Thu, 5 Dec 2024 22:07:19 +0000 Subject: [PATCH] Format Signed-off-by: mgoin --- tests/entrypoints/llm/test_guided_generate.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/entrypoints/llm/test_guided_generate.py b/tests/entrypoints/llm/test_guided_generate.py index 4de2f0596bfea..ed50ec6bbc9eb 100644 --- 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}"