Skip to content

Commit 83243d4

Browse files
author
Mohamed Zeidan
committed
jumpstart public hub fix
1 parent dc119db commit 83243d4

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

src/sagemaker/hyperpod/inference/jumpstart_public_hub_visualization_utils.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -217,10 +217,6 @@ def _generate_deployment_config(self, model_id):
217217
instance_type = default_type if default_type else '\<ENTER-INSTANCE-TYPE\>'
218218
types_comment = self._format_instance_types_comment(supported_types)
219219

220-
# Generate timestamp for unique endpoint name
221-
timestamp = str(int(time.time()))
222-
endpoint_name_value = f"{model_id}-{timestamp}"
223-
224220
config_code = f'''# Deployment configuration for {model_id}
225221
from sagemaker.hyperpod.inference.config.hp_jumpstart_endpoint_config import (
226222
Model, Server, SageMakerEndpoint
@@ -236,7 +232,9 @@ def _generate_deployment_config(self, model_id):
236232
server = Server(
237233
instance_type='{instance_type}',
238234
)
239-
endpoint_name = SageMakerEndpoint(name='{endpoint_name_value}')
235+
236+
# Default endpoint name using model_id, modify as desired
237+
endpoint_name = SageMakerEndpoint(name='{model_id}')
240238
241239
# Create endpoint spec
242240
js_endpoint = HPJumpStartEndpoint(
@@ -300,4 +298,4 @@ def _style_dataframe(df):
300298

301299
def _get_table_layout(data_length):
302300
"""Get appropriate table layout based on data size."""
303-
return {} if data_length > 10 else {"topStart": None, "topEnd": "search"}
301+
return {} if data_length > 10 else {"topStart": None, "topEnd": "search"}

0 commit comments

Comments
 (0)