Skip to content

Commit b7a8216

Browse files
dacorvoclaude
andcommitted
fix(test): unify greedy expected output across platforms
The Eiffel Tower generation is "one of" on all platforms (trn1, inf2, trn2), not just trn2. Remove the platform-dependent branch that caused CI failures on non-trn2 runners. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 8f69e12 commit b7a8216

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/vllm/engine/test_vllm_engine_generate.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from vllm import LLM, RequestOutput, SamplingParams
1111
from vllm.platforms import current_platform
1212

13-
from optimum.neuron.utils.instance import current_instance_type
1413
from optimum.neuron.vllm.platform import OptimumNeuronPlatform
1514

1615

@@ -95,10 +94,9 @@ def test_vllm_greedy_expectations(neuron_llm_config: dict[str, Any]):
9594

9695
outputs = llm.generate(prompts, sampling_params)
9796

98-
trn2 = current_instance_type() == "trn2"
9997
expected_outputs = [
10098
" the head of state and government of the United States",
101-
" Paris. The Eiffel Tower is located in Paris. The Eiffel Tower is " + ("one of" if trn2 else "a famous"),
99+
" Paris. The Eiffel Tower is located in Paris. The Eiffel Tower is one of",
102100
" The world was holding its breath as the world's top scientists and engineers gathered at the secret underground facility to witness the unveiling of the ultimate time machine.",
103101
" to find happiness and fulfillment in the present moment. It's a simple yet profound concept that can bring joy and peace to our lives.\n\nAs I reflect on my own life, I realize that I've",
104102
" blue, but what about the colour of the sky",

0 commit comments

Comments
 (0)