Skip to content

Commit c0c8fd4

Browse files
fix sm test commit id issue
1 parent 5e79a45 commit c0c8fd4

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

test/vllm/trigger_test.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import os, sys
22
import logging
33
from typing import List
4+
import uuid
45

56
from test.test_utils import get_dlc_images, is_pr_context
67
from test.vllm.eks.eks_test import test_vllm_on_eks
@@ -16,7 +17,7 @@
1617
SM_TEST = os.path.join("sagemaker", "test_sm_endpoint.py")
1718

1819

19-
def run_platform_tests(platform: str, images: List[str], commit_id: str):
20+
def run_platform_tests(platform: str, images: List[str]):
2021
"""
2122
Run tests for a specific platform
2223
"""
@@ -47,7 +48,8 @@ def run_platform_tests(platform: str, images: List[str], commit_id: str):
4748
elif platform == "sagemaker":
4849
LOGGER.info("Running sagemaker test")
4950
try:
50-
endpoint_name = f"test-sm-vllm-endpoint-{commit_id}"
51+
TEST_ID = str(uuid.uuid4())
52+
endpoint_name = f"test-sm-vllm-endpoint-{TEST_ID}"
5153
test_vllm_on_sagemaker(images[0], endpoint_name)
5254

5355
except Exception as e:
@@ -75,7 +77,7 @@ def test():
7577
standard_images_list = [image_uri for image_uri in all_image_list if "example" not in image_uri]
7678
LOGGER.info(f"\nImages URIs:\n{standard_images_list}")
7779

78-
run_platform_tests(platform=test_type, images=standard_images_list, commit_id=commit_id)
80+
run_platform_tests(platform=test_type, images=standard_images_list)
7981

8082

8183
if __name__ == "__main__":

0 commit comments

Comments
 (0)