Merge pull request #45 from zhaoqizqwang/add-inference-classes

Add tls config

Author: mollyheamazon

sagemaker-hyperpod/src/sagemaker/hyperpod/inference/hp_endpoint.py

@@ -8,6 +8,7 @@
     ModelInvocationPort,
     ModelVolumeMount,
     Resources,
+    TlsConfig,
 )
 from sagemaker.hyperpod.inference.hp_endpoint_base import HPEndpointBase
 from datetime import datetime
@@ -137,6 +138,7 @@ def create(
         endpoint_name: str = None,
         model_volume_mount_name: str = None,
         model_volume_mount_path: str = None,
+        tls_s3_uri: str = None,
     ):
         instance = cls()
 
@@ -188,6 +190,10 @@ def create(
             resources=Resources(),
         )
 
+        tls_config = TlsConfig(
+            tls_certificate_output_s3_uri=tls_s3_uri,
+        )
+
         # create spec config
         spec = InferenceEndpointConfigSpec(
             instance_type=instance_type,
@@ -196,6 +202,7 @@ def create(
             model_source_config=model_source_config,
             worker=worker,
             endpoint_name=endpoint_name,
+            tls_config=tls_config,
         )
 
         instance.call_create_api(
@@ -248,9 +255,10 @@ def list_endpoints(
         )
 
         output_data = []
-        for item in response["items"]:
-            metadata = item["metadata"]
-            output_data.append((metadata["name"], metadata["creationTimestamp"]))
+        if response and response["items"]:
+            for item in response["items"]:
+                metadata = item["metadata"]
+                output_data.append((metadata["name"], metadata["creationTimestamp"]))
         headers = ["METADATA NAME", "CREATE TIME"]
 
         print(tabulate(output_data, headers=headers))