Skip to content

Commit 903d4fd

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 8c11af8 commit 903d4fd

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/model_serving/model_runtime/vllm/speculative_decoding/test_granite_7b_lab_draft.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,9 @@
4747
indirect=True,
4848
)
4949
class TestGraniteLabDraftModel:
50-
@pytest.mark.xfail(reason="vLLM does not support MLPSpeculatorPreTrainedModel architecture for draft model speculative decoding")
50+
@pytest.mark.xfail(
51+
reason="vLLM does not support MLPSpeculatorPreTrainedModel architecture for draft model speculative decoding"
52+
)
5153
def test_spec_draft_inference(
5254
self,
5355
vllm_inference_service: Generator[InferenceService, Any, Any],
@@ -97,7 +99,9 @@ def test_spec_draft_inference(
9799
indirect=True,
98100
)
99101
class TestMultiGraniteLabDraftModel:
100-
@pytest.mark.xfail(reason="vLLM does not support MLPSpeculatorPreTrainedModel architecture for draft model speculative decoding")
102+
@pytest.mark.xfail(
103+
reason="vLLM does not support MLPSpeculatorPreTrainedModel architecture for draft model speculative decoding"
104+
)
101105
def test_multi_spec_draft_inference(
102106
self,
103107
vllm_inference_service: Generator[InferenceService, Any, Any],

0 commit comments

Comments
 (0)