Fix list_pods and AZ_ID error message (#238)

zhaoqizqwang · web-flow · commit 12730ca5ae1c · 2025-08-26T15:46:56.000-07:00
Tested locally by running the commands and saw exptected results
diff --git a/src/sagemaker/hyperpod/cli/init_utils.py b/src/sagemaker/hyperpod/cli/init_utils.py
@@ -517,7 +517,7 @@ def add_default_az_ids_to_config(dir_path: str, region: str):
             az_ids = all_az_ids[:2]
 
             update_list_field_in_config(dir_path, 'availability_zone_ids', az_ids)
-            click.secho(f"No availability_zone_ids provided. Using default AZ Id: az_ids.", fg="yellow")
+            click.secho(f"No availability_zone_ids provided. Using default AZ Id: {az_ids}.", fg="yellow")
         except Exception as e:
             raise Exception(f"Failed to find default availability_zone_ids for region {region}. Please provide one in config.yaml. Error details: {e}")
 
diff --git a/src/sagemaker/hyperpod/inference/hp_jumpstart_endpoint.py b/src/sagemaker/hyperpod/inference/hp_jumpstart_endpoint.py
@@ -264,7 +264,7 @@ def list_pods(cls, namespace=None, endpoint_name=None):
             endpoints.add(endpoint_name)
         else:
             list_response = cls.call_list_api(
-                kind=INFERENCE_ENDPOINT_CONFIG_KIND,
+                kind=JUMPSTART_MODEL_KIND,
                 namespace=namespace,
             )
             if list_response and list_response["items"]:

Original file line number	Diff line number	Diff line change
`@@ -264,7 +264,7 @@ def list_pods(cls, namespace=None, endpoint_name=None):`
`264`	`264`	`endpoints.add(endpoint_name)`
`265`	`265`	`else:`
`266`	`266`	`list_response = cls.call_list_api(`
`267`		`- kind=INFERENCE_ENDPOINT_CONFIG_KIND,`
	`267`	`+ kind=JUMPSTART_MODEL_KIND,`
`268`	`268`	`namespace=namespace,`
`269`	`269`	`)`
`270`	`270`	`if list_response and list_response["items"]:`