fix pre-commit error

Signed-off-by: csy1204 <[email protected]>

Author: csy1204

tests/entrypoints/llm/test_guided_generate.py

@@ -388,13 +388,26 @@ def test_guided_json_completion_with_enum(llm, guided_decoding_backend: str):
 
 @pytest.mark.skip_global_cleanup
 @pytest.mark.parametrize("guided_decoding_backend", GUIDED_DECODING_BACKENDS)
-def test_guided_number_range_json_completion(llm, guided_decoding_backend: str):
+def test_guided_number_range_json_completion(llm,
+                                             guided_decoding_backend: str):
     sample_number_range_schema = {
         "type": "object",
         "properties": {
-            "age": {"type": "integer", "minimum": 18, "maximum": 99},
-            "score": {"type": "number", "minimum": 0.0, "maximum": 100.0},
-            "level": {"type": "integer", "minimum": 1, "maximum": 10},
+            "age": {
+                "type": "integer",
+                "minimum": 18,
+                "maximum": 99
+            },
+            "score": {
+                "type": "number",
+                "minimum": 0.0,
+                "maximum": 100.0
+            },
+            "level": {
+                "type": "integer",
+                "minimum": 1,
+                "maximum": 10
+            },
         },
         "required": ["age", "score", "level"]
     }
@@ -420,7 +433,8 @@ def test_guided_number_range_json_completion(llm, guided_decoding_backend: str):
         assert generated_text is not None
         print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}")
         output_json = json.loads(generated_text)
-        jsonschema.validate(instance=output_json, schema=sample_number_range_schema)
+        jsonschema.validate(instance=output_json,
+                            schema=sample_number_range_schema)
         assert 18 <= output_json["age"] <= 99
         assert 0.0 <= output_json["score"] <= 100.0
         assert 1 <= output_json["level"] <= 10
@@ -432,7 +446,10 @@ def test_guided_pattern_json_completion(llm, guided_decoding_backend: str):
     sample_pattern_schema = {
         "type": "object",
         "properties": {
-            "zipcode": {"type": "string", "pattern": r"^\\d{5}(-\\d{4})?$"},
+            "zipcode": {
+                "type": "string",
+                "pattern": r"^\\d{5}(-\\d{4})?$"
+            },
         },
         "required": ["zipcode"]
     }
@@ -441,11 +458,10 @@ def test_guided_pattern_json_completion(llm, guided_decoding_backend: str):
                                      guided_decoding=GuidedDecodingParams(
                                          json=sample_pattern_schema,
                                          backend=guided_decoding_backend))
-    outputs = llm.generate(prompts=[
-        "Create a JSON object for a US zipcode (5 or 9 digits)."
-    ] * 2,
-                           sampling_params=sampling_params,
-                           use_tqdm=True)
+    outputs = llm.generate(
+        prompts=["Create a JSON object for a US zipcode (5 or 9 digits)."] * 2,
+        sampling_params=sampling_params,
+        use_tqdm=True)
 
     assert outputs is not None
 
@@ -459,4 +475,5 @@ def test_guided_pattern_json_completion(llm, guided_decoding_backend: str):
         print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}")
         output_json = json.loads(generated_text)
         jsonschema.validate(instance=output_json, schema=sample_pattern_schema)
-        assert re.fullmatch(r"^\d{5}(-\d{4})?$", output_json["zipcode"]) is not None
+        assert re.fullmatch(r"^\d{5}(-\d{4})?$",
+                            output_json["zipcode"]) is not None

tests/v1/entrypoints/llm/test_struct_output_generate.py

@@ -425,9 +425,21 @@ def test_structured_output_number_range(
     number_range_schema = {
         "type": "object",
         "properties": {
-            "age": {"type": "integer", "minimum": 18, "maximum": 99},
-            "score": {"type": "number", "minimum": 0.0, "maximum": 100.0},
-            "level": {"type": "integer", "minimum": 1, "maximum": 10},
+            "age": {
+                "type": "integer",
+                "minimum": 18,
+                "maximum": 99
+            },
+            "score": {
+                "type": "number",
+                "minimum": 0.0,
+                "maximum": 100.0
+            },
+            "level": {
+                "type": "integer",
+                "minimum": 1,
+                "maximum": 10
+            },
         },
         "required": ["age", "score", "level"]
     }
@@ -475,19 +487,21 @@ def test_structured_output_pattern(
     pattern_schema = {
         "type": "object",
         "properties": {
-            "zipcode": {"type": "string", "pattern": r"^\\d{5}(-\\d{4})?$"},
+            "zipcode": {
+                "type": "string",
+                "pattern": r"^\\d{5}(-\\d{4})?$"
+            },
         },
         "required": ["zipcode"]
     }
     sampling_params = SamplingParams(
         temperature=1.0,
         max_tokens=1000,
         guided_decoding=GuidedDecodingParams(json=pattern_schema))
-    outputs = llm.generate(prompts=[
-        "Create a JSON object for a US zipcode (5 or 9 digits)."
-    ] * 2,
-                           sampling_params=sampling_params,
-                           use_tqdm=True)
+    outputs = llm.generate(
+        prompts=["Create a JSON object for a US zipcode (5 or 9 digits)."] * 2,
+        sampling_params=sampling_params,
+        use_tqdm=True)
 
     assert outputs is not None
     for output in outputs:
@@ -499,4 +513,5 @@ def test_structured_output_pattern(
             assert "\n" not in generated_text
         output_json = json.loads(generated_text)
         jsonschema.validate(instance=output_json, schema=pattern_schema)
-        assert re.fullmatch(r"^\d{5}(-\d{4})?$", output_json["zipcode"]) is not None
+        assert re.fullmatch(r"^\d{5}(-\d{4})?$",
+                            output_json["zipcode"]) is not None