enable vLLM EKS EC2 test on pipeline

test/testrunner.py

@@ -32,7 +32,7 @@
 )
 from test_utils import KEYS_TO_DESTROY_FILE
 from test_utils.pytest_cache import PytestCache
-from test.vllm.trigger_test import test
+from test.vllm.trigger_test import test as test_vllm
 
 from src.codebuild_environment import get_codebuild_project_name
 
@@ -323,6 +323,7 @@ def main():
             "functionality_sanity",
             "security_sanity",
             "eks",
+            "ec2",
         }:
             LOGGER.info(
                 f"NOTE: {specific_test_type} tests not supported on vllm images. Skipping..."
@@ -426,7 +427,7 @@ def main():
             if framework == "vllm":
                 try:
                     LOGGER.info(f"Running vLLM EKS EC2 tests with image: {all_image_list[0]}")
-                    test()
+                    test_vllm()
                     LOGGER.info("vLLM EKS EC2 tests completed successfully")
                     # Exit function after vLLM tests
                     return

test/vllm/eks/eks_test.py

@@ -21,7 +21,8 @@
 LOGGER.addHandler(logging.StreamHandler(sys.stdout))
 
 AWS_REGION = "us-west-2"
-CLUSTER_NAME = "dlc-vllm-PR"
+build_context = os.getenv("BUILD_CONTEXT")
+CLUSTER_NAME = f"dlc-vllm-{build_context}"
 VLLM_NAMESPACE = "vllm"
 TEST_DIR = os.path.dirname(os.path.abspath(__file__))
 LWS_YAML = os.path.join(TEST_DIR, "test_artifacts", "vllm-deepseek-32b-lws.yaml")