change test

richardhuo-nv · richardhuo-nv · commit 071620cbaaaa · 2025-03-03T23:17:53.000-08:00
diff --git a/tests/test_model_repository.py b/tests/test_model_repository.py
@@ -64,7 +64,7 @@ def test_can_get_tensorrtllm_engine_folder_from_model_repository(self):
         ), f"engine path found is not as expected. Expected: {expected_engine_path}. Found: {trtllm_utils.get_engine_path()}"
 
     @pytest.mark.skipif(
-        os.environ.get("IMAGE_KIND") != "TRTLLM", reason="Only run for VLLM image"
+        os.environ.get("IMAGE_KIND") != "TRTLLM", reason="Only run for TRT-LLM image"
     )
     @pytest.mark.timeout(DOWNLOAD_TIMEOUT_SECS)
     def test_can_get_llmapi_model_id_from_model_repository(self):

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ def test_can_get_tensorrtllm_engine_folder_from_model_repository(self):`
`64`	`64`	`), f"engine path found is not as expected. Expected: {expected_engine_path}. Found: {trtllm_utils.get_engine_path()}"`
`65`	`65`
`66`	`66`	`@pytest.mark.skipif(`
`67`		`- os.environ.get("IMAGE_KIND") != "TRTLLM", reason="Only run for VLLM image"`
	`67`	`+ os.environ.get("IMAGE_KIND") != "TRTLLM", reason="Only run for TRT-LLM image"`
`68`	`68`	`)`
`69`	`69`	`@pytest.mark.timeout(DOWNLOAD_TIMEOUT_SECS)`
`70`	`70`	`def test_can_get_llmapi_model_id_from_model_repository(self):`