Fix issues

Author: Jonathan Makunga

src/sagemaker/serve/utils/optimize_utils.py

@@ -15,7 +15,6 @@
 
 import re
 import logging
-import uuid
 from typing import Dict, Any, Optional, Union, List
 
 from sagemaker import Model
@@ -169,7 +168,7 @@ def _generate_channel_name(additional_model_data_sources: Optional[List[Dict]])
     Returns:
         str: The channel name.
     """
-    channel_name = f"model-builder-channel-{uuid.uuid4().hex}"
+    channel_name = "model-builder-channel"
     if additional_model_data_sources and len(additional_model_data_sources) > 0:
         channel_name = additional_model_data_sources[0].get("ChannelName", channel_name)
 

tests/unit/sagemaker/jumpstart/constants.py

@@ -7516,6 +7516,8 @@
     "inference_config_rankings": None,
     "training_config_rankings": None,
     "hosting_additional_data_sources": None,
+    "hosting_neuron_model_id": None,
+    "hosting_neuron_model_version": None,
 }
 
 BASE_HOSTING_ADDITIONAL_DATA_SOURCES = {