From b08ce26ac101c02c9e757d44efcf905ae2ba7971 Mon Sep 17 00:00:00 2001 From: trtllm-agent <296075020+trtllm-agent@users.noreply.github.com> Date: Mon, 6 Jul 2026 13:54:08 -0700 Subject: [PATCH] [nvbugs/6418017][fix] Replace Triton test_medusa with cleanup assertion PR #15763 removed examples/medusa/ as part of the legacy TensorRT-backend cleanup, but tests/integration/defs/triton_server/build_model.sh still ran 'pushd examples/medusa' for the medusa target, causing the Triton smoke test to fail with 'pushd: examples/medusa: No such file or directory'. Replace the body of test_medusa with a lightweight assertion that examples/medusa/ remains absent under llm_root. This keeps the pytest node id resolvable (the harness runs the test by exact node id) while removing the broken build/test.sh invocation. The dead 'medusa' branches in build_model.sh and test.sh are no longer reachable and can be pruned in a follow-up cleanup. Signed-off-by: trtllm-agent <296075020+trtllm-agent@users.noreply.github.com> --- .../integration/defs/triton_server/test_triton.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) 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)