Adds more debug statements

Author: mrDzurb

ads/aqua/app.py

@@ -339,6 +339,9 @@ def get_config(
         config_file_path = os.path.join(config_path, config_file_name)
         if is_path_exists(config_file_path):
             try:
+                logger.info(
+                    f"Loading config: `{config_file_name}` from `{config_path}`"
+                )
                 config = load_config(
                     config_path,
                     config_file_name=config_file_name,
@@ -378,9 +381,11 @@ def build_cli(self) -> str:
         """
         cmd = f"ads aqua {self._command}"
         params = [
-            f"--{field.name} {json.dumps(getattr(self, field.name))}"
-            if isinstance(getattr(self, field.name), dict)
-            else f"--{field.name} {getattr(self, field.name)}"
+            (
+                f"--{field.name} {json.dumps(getattr(self, field.name))}"
+                if isinstance(getattr(self, field.name), dict)
+                else f"--{field.name} {getattr(self, field.name)}"
+            )
             for field in fields(self.__class__)
             if getattr(self, field.name) is not None
         ]

ads/aqua/common/utils.py

@@ -252,7 +252,7 @@ def load_config(file_path: str, config_file_name: str, **kwargs) -> dict:
     return config
 
 
-def list_os_files_with_extension(oss_path: str, extension: str) -> [str]:
+def list_os_files_with_extension(oss_path: str, extension: str) -> List[str]:
     """
     List files in the specified directory with the given extension.
 

ads/aqua/modeldeployment/utils.py

@@ -62,7 +62,10 @@ def load(
             cannot be determined, an appropriate error message is included in the summary.
         """
         # Fetch deployment configurations concurrently.
+        logger.debug(f"Loading model deployment configuration for models: {model_ids}")
         deployment_configs = self._fetch_deployment_configs_concurrently(model_ids)
+
+        logger.debug(f"Loaded config: {deployment_configs}")
         model_shape_gpu, deployment = self._extract_model_shape_gpu(deployment_configs)
 
         # Initialize the summary result with the deployment configurations.
@@ -80,6 +83,8 @@ def load(
 
         # Identify common deployment shapes among all models.
         common_shapes = self._get_common_shapes(model_shape_gpu)
+        logger.debug(f"Common Shapes: {common_shapes} from: {model_shape_gpu}")
+
         if not common_shapes:
             summary.error_message = (
                 "The selected models do not share any common deployment shapes. "
@@ -94,6 +99,9 @@ def load(
         gpu_allocation = self._compute_gpu_allocation(
             common_shapes, model_shape_gpu, primary_model_id
         )
+
+        logger.debug(f"GPU Allocation: {gpu_allocation}")
+
         if not gpu_allocation:
             summary.error_message = (
                 "Unable to determine a valid GPU allocation for the selected models based on their current configurations. "