Skip to content

Commit dce4249

Browse files
authored
Fix integration tests (#2947)
1 parent 271602b commit dce4249

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

engines/python/setup/djl_python/properties_manager/vllm_rb_properties.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def validate_pipeline_parallel(self):
122122
def validate_tool_call_parser(self):
123123
if self.enable_auto_tool_choice:
124124
from vllm.entrypoints.openai.tool_parsers import ToolParserManager
125-
valid_tool_parses = ToolParserManager.tool_parsers.keys()
125+
valid_tool_parses = ToolParserManager.list_registered()
126126
if self.tool_call_parser not in valid_tool_parses:
127127
raise ValueError(
128128
f"Invalid tool call parser: {self.tool_call_parser} "
@@ -133,8 +133,7 @@ def validate_tool_call_parser(self):
133133
def validate_reasoning_parser(self):
134134
if self.enable_reasoning:
135135
from vllm.reasoning.abs_reasoning_parsers import ReasoningParserManager
136-
valid_reasoning_parses = ReasoningParserManager.reasoning_parsers.keys(
137-
)
136+
valid_reasoning_parses = ReasoningParserManager.list_registered()
138137
if self.reasoning_parser not in valid_reasoning_parses:
139138
raise ValueError(
140139
f"Invalid reasoning parser: {self.reasoning_parser} "

tests/integration/tests.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -790,8 +790,7 @@ class TestVllm_p4d:
790790
def test_qwen3_vl_32b_instruct(self):
791791
with Runner('lmi', 'qwen3-vl-32b-instruct') as r:
792792
prepare.build_vllm_async_model("qwen3-vl-32b-instruct")
793-
env = ["VLLM_ATTENTION_BACKEND=TORCH_SDPA"]
794-
r.launch(env_vars=env)
793+
r.launch()
795794
client.run("multimodal qwen3-vl-32b-instruct".split())
796795

797796
def test_llama_4_scout_17b_16e_instruct(self):

0 commit comments

Comments
 (0)