Add unique time string to integ test (#150)

* Add unique time string to integ test

* Update syntax

Author: zhaoqizqwang

test/integration_tests/inference/cli/test_cli_custom_fsx_inference.py

@@ -14,6 +14,7 @@
     custom_list_pods
 )
 from sagemaker.hyperpod.inference.hp_endpoint import HPEndpoint
+from test.integration_tests.utils import get_time_str
 
 # --------- Test Configuration ---------
 NAMESPACE = "integration"
@@ -36,7 +37,7 @@ def runner():
 
 @pytest.fixture(scope="module")
 def custom_endpoint_name():
-    return f"custom-cli-integration-fsx"
+    return "custom-cli-integration-fsx-" + get_time_str()
 
 @pytest.fixture(scope="module")
 def sagemaker_client():

test/integration_tests/inference/cli/test_cli_custom_s3_inference.py

@@ -1,5 +1,4 @@
 import time
-import uuid
 import pytest
 import boto3
 import os
@@ -14,6 +13,7 @@
     custom_list_pods
 )
 from sagemaker.hyperpod.inference.hp_endpoint import HPEndpoint
+from test.integration_tests.utils import get_time_str
 
 # --------- Test Configuration ---------
 NAMESPACE = "integration"
@@ -36,7 +36,7 @@ def runner():
 
 @pytest.fixture(scope="module")
 def custom_endpoint_name():
-    return f"custom-cli-integration-s3"
+    return "custom-cli-integration-s3-" + get_time_str()
 
 @pytest.fixture(scope="module")
 def sagemaker_client():

test/integration_tests/inference/cli/test_cli_jumpstart_inference.py

@@ -1,12 +1,12 @@
 import time
-import uuid
 import pytest
 import boto3
 from click.testing import CliRunner
 from sagemaker.hyperpod.cli.commands.inference import (
     js_create, custom_invoke, js_list, js_describe, js_delete, js_get_operator_logs, js_list_pods
 )
 from sagemaker.hyperpod.inference.hp_jumpstart_endpoint import HPJumpStartEndpoint
+from test.integration_tests.utils import get_time_str
 
 # --------- Test Configuration ---------
 NAMESPACE = "integration"
@@ -21,7 +21,7 @@ def runner():
 
 @pytest.fixture(scope="module")
 def js_endpoint_name():
-    return f"js-cli-integration"
+    return "js-cli-integration-" + get_time_str()
 
 @pytest.fixture(scope="module")
 def sagemaker_client():

test/integration_tests/inference/sdk/test_sdk_custom_fsx_inference.py

@@ -1,21 +1,19 @@
 import time
-import uuid
-import json
 import pytest
 import boto3
 import os
 from sagemaker.hyperpod.inference.hp_endpoint import HPEndpoint
 from sagemaker.hyperpod.inference.config.hp_endpoint_config import (
     ModelSourceConfig, FsxStorage, TlsConfig, Worker, ModelVolumeMount,
-    ModelInvocationPort, Resources, EnvironmentVariables, AutoScalingSpec,
-    CloudWatchTrigger, Dimensions, Metrics
+    ModelInvocationPort, Resources, EnvironmentVariables,
 )
 import sagemaker_core.main.code_injection.codec as codec
+from test.integration_tests.utils import get_time_str
 
 # --------- Test Configuration ---------
 NAMESPACE = "integration"
 REGION = "us-east-2"
-ENDPOINT_NAME = f"custom-sdk-integration-fsx"
+ENDPOINT_NAME = "custom-sdk-integration-fsx-" + get_time_str()
 
 MODEL_NAME = f"test-model-integration-sdk-fsx"
 MODEL_LOCATION = "hf-eqa"

test/integration_tests/inference/sdk/test_sdk_custom_s3_inference.py

@@ -1,21 +1,19 @@
 import time
-import uuid
-import json
 import pytest
 import boto3
 import os
 from sagemaker.hyperpod.inference.hp_endpoint import HPEndpoint
 from sagemaker.hyperpod.inference.config.hp_endpoint_config import (
     ModelSourceConfig, S3Storage, TlsConfig, Worker, ModelVolumeMount,
-    ModelInvocationPort, Resources, EnvironmentVariables, AutoScalingSpec,
-    CloudWatchTrigger, Dimensions, Metrics
+    ModelInvocationPort, Resources, EnvironmentVariables
 )
 import sagemaker_core.main.code_injection.codec as codec
+from test.integration_tests.utils import get_time_str
 
 # --------- Test Configuration ---------
 NAMESPACE = "integration"
 REGION = "us-east-2"
-ENDPOINT_NAME = f"custom-sdk-integration-s3"
+ENDPOINT_NAME = "custom-sdk-integration-s3-" + get_time_str()
 
 MODEL_NAME = f"test-model-integration-sdk-s3"
 MODEL_LOCATION = "hf-eqa"

test/integration_tests/inference/sdk/test_sdk_jumpstart_inference.py

@@ -1,19 +1,17 @@
 import time
-import uuid
-import json
 import pytest
 import boto3
-
 from sagemaker.hyperpod.inference.hp_jumpstart_endpoint import HPJumpStartEndpoint
 from sagemaker.hyperpod.inference.config.hp_jumpstart_endpoint_config import (
-    Model, Server, SageMakerEndpoint, TlsConfig
+    Model, Server, SageMakerEndpoint
 )
 import sagemaker_core.main.code_injection.codec as codec
+from test.integration_tests.utils import get_time_str
 
 # --------- Config ---------
 NAMESPACE = "integration"
 REGION = "us-east-2"
-ENDPOINT_NAME = "js-sdk-integration"
+ENDPOINT_NAME = "js-sdk-integration-" + get_time_str()
 
 INSTANCE_TYPE = "ml.g5.4xlarge"
 MODEL_ID = "deepseek-llm-r1-distill-qwen-1-5b"

test/integration_tests/utils.py

@@ -1,5 +1,6 @@
 import subprocess
 import logging
+import datetime
 
 logger = logging.getLogger(__name__)
 
@@ -18,3 +19,7 @@ def execute_command(command):
         logger.error(f"Stdout: {e.stdout}")
         logger.error(f"Stderr: {e.stderr}")
         raise RuntimeError(f"Failed to execute command: {' '.join(command)}. Error: {e}")
+
+def get_time_str():
+    now = datetime.datetime.now()
+    return now.strftime("%m%d-%H%M%S")