diff --git a/tests/integration/defs/triton_server/test_triton.py b/tests/integration/defs/triton_server/test_triton.py index 6d710ad5ba1d..2a837d6b6436 100644 --- a/tests/integration/defs/triton_server/test_triton.py +++ b/tests/integration/defs/triton_server/test_triton.py @@ -292,13 +292,13 @@ def test_gpt_gather_logits(tritonserver_test_root, test_name, llm_root, @pytest.mark.parametrize("test_name", ["medusa"], indirect=True) -def test_medusa(tritonserver_test_root, test_name, llm_root, model_path, - engine_dir): - build_model(test_name, llm_root, tritonserver_test_root) - tokenizer_type = "llama" - run_shell_command( - f"cd {tritonserver_test_root} && ./test.sh {test_name} {engine_dir} {model_path} {tokenizer_type}", - llm_root) +def test_medusa(tritonserver_test_root, test_name, llm_root): + # examples/medusa was removed with the legacy TensorRT-backend cleanup; + # guard against silent reintroduction. + medusa_example_dir = os.path.join(llm_root, "examples", "medusa") + assert not os.path.exists(medusa_example_dir), ( + f"{medusa_example_dir} was reintroduced; update this test or restore " + "the Triton medusa build pipeline in build_model.sh and test.sh.") @pytest.mark.parametrize("test_name", ["eagle"], indirect=True)