Updated pr.

Author: lu-ohai

ads/aqua/modeldeployment/deployment.py

@@ -162,7 +162,7 @@ def create(
         # Create an AquaModelApp instance once to perform the deployment creation.
         model_app = AquaModelApp()
         if create_deployment_details.model_id:
-            logger.info(
+            logger.debug(
                 f"Single model ({create_deployment_details.model_id}) provided. "
                 "Delegating to single model creation method."
             )
@@ -250,7 +250,7 @@ def create(
                         f"Only the following container families are supported: {supported_container_families}."
                     )
 
-            logger.info(
+            logger.debug(
                 f"Multi models ({model_ids}) provided. Delegating to multi model creation method."
             )
 
@@ -1066,11 +1066,11 @@ def get_multimodel_deployment_config(
         )
 
         if len(model_ids) == 1:
-            return multi_model_deployment_config_loader.load_single(
+            return multi_model_deployment_config_loader.load_model_deployment_configuration(
                 shapes=available_shapes, model_id=model_ids[0]
             )
 
-        return multi_model_deployment_config_loader.load(
+        return multi_model_deployment_config_loader.load_multi_model_deployment_configuration(
             shapes=available_shapes,
             model_ids=model_ids,
             primary_model_id=primary_model_id,

ads/aqua/modeldeployment/utils.py

@@ -44,7 +44,7 @@ def __init__(self, deployment_app: AquaApp):
         """
         self.deployment_app = deployment_app
 
-    def load(
+    def load_multi_model_deployment_configuration(
         self,
         shapes: List[ComputeShapeSummary],
         model_ids: List[str],
@@ -132,7 +132,7 @@ def load(
         summary.gpu_allocation = gpu_allocation
         return summary
 
-    def load_single(
+    def load_model_deployment_configuration(
         self,
         shapes: List[ComputeShapeSummary],
         model_id: str,
@@ -160,15 +160,15 @@ def load_single(
         ]
 
         deployment = {
-            model_id: {
-                "shape": [shape.upper() for shape in deployment_config.shape],
-                "configuration": {
+            model_id: AquaDeploymentConfig(
+                shape=[shape.upper() for shape in deployment_config.shape],
+                configuration={
                     shape.upper(): deployment_config.configuration.get(
                         shape, ConfigurationItem()
                     )
                     for shape in deployment_config.shape
                 },
-            }
+            )
         }
 
         # Initialize the summary result with the deployment configurations.