Fix get job details for CLI flow (#104)

Aditi2424 · adishaa · web-flow · commit 9ff002f949bc · 2025-06-18T23:31:54.000-07:00
Co-authored-by: adishaa &lt;adishaa@amazon.com&gt;
diff --git a/src/hyperpod_cli/commands/job.py b/src/hyperpod_cli/commands/job.py
@@ -767,6 +767,11 @@ def start_job(
                 if custom_labels
                 else None,
             )
+
+            if "custom_labels" not in config["cluster"]["cluster_config"] or config["cluster"]["cluster_config"]["custom_labels"] is None:
+                config["cluster"]["cluster_config"]["custom_labels"] = {}
+            config["cluster"]["cluster_config"]["custom_labels"][USER_NAME_LABEL_KEY] = user_name
+
             _override_or_remove(
                 config["cluster"]["cluster_config"], "priority_class_name", priority
             )

Original file line number	Diff line number	Diff line change
`@@ -767,6 +767,11 @@ def start_job(`
`767`	`767`	`if custom_labels`
`768`	`768`	`else None,`
`769`	`769`	`)`
	`770`	`+`
	`771`	`+ if "custom_labels" not in config["cluster"]["cluster_config"] or config["cluster"]["cluster_config"]["custom_labels"] is None:`
	`772`	`+ config["cluster"]["cluster_config"]["custom_labels"] = {}`
	`773`	`+ config["cluster"]["cluster_config"]["custom_labels"][USER_NAME_LABEL_KEY] = user_name`
	`774`	`+`
`770`	`775`	`_override_or_remove(`
`771`	`776`	`config["cluster"]["cluster_config"], "priority_class_name", priority`
`772`	`777`	`)`