formatting

vllm-project · Jul 6, 2024 · e978360 · e978360
1 parent e79ffa8
commit e978360
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/tests/distributed/test_distributed_gptq_marlin.py b/tests/distributed/test_distributed_gptq_marlin.py
@@ -31,11 +31,9 @@ def test_models(vllm_runner, example_prompts, model, dtype: str,
         pytest.skip(
             f"Need at least {tensor_parallel_size} GPUs to run the test.")
 
-
     if not is_quant_method_supported("gptq_marlin"):
-        pytest.skip(
-            "gptq_marlin is not supported on this GPU type.")
-
+        pytest.skip("gptq_marlin is not supported on this GPU type.")
+
     distributed_executor_backend = os.getenv("DISTRIBUTED_EXECUTOR_BACKEND")
     run_test(vllm_runner,
              example_prompts,

diff --git a/tests/models/test_gptq_marlin.py b/tests/models/test_gptq_marlin.py
@@ -110,8 +110,7 @@ def test_models(
     num_logprobs: int,
 ) -> None:
     if not is_quant_method_supported("gptq_marlin"):
-        pytest.skip(
-            "gptq_marlin is not supported on this GPU type.")
+        pytest.skip("gptq_marlin is not supported on this GPU type.")
     run_test(
         vllm_runner,
         example_prompts,