Fix adding necessary args and cluster credentials

AI-Hypercomputer · Oct 15, 2024 · aff257b · aff257b
1 parent 42f92f9
commit aff257b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/xpk/commands/workload.py b/src/xpk/commands/workload.py
@@ -324,6 +324,9 @@ def workload_create(args) -> None:
   Returns:
     0 if successful and 1 otherwise.
   """
+  add_zone_and_project(args)
+  get_cluster_credentials(args)
+
   if args.headless and not is_cluster_using_clouddns(args):
     xpk_print(
         'Please run xpk cluster create-pathways first, to upgrade and enable'

diff --git a/src/xpk/core/core.py b/src/xpk/core/core.py
@@ -201,7 +201,7 @@ def add_zone_and_project(args):
     args.project = get_project()
   if not args.zone:
     args.zone = get_zone()
-  xpk_print(f'Working on {args.project=} and {args.zone}')
+  xpk_print(f'Working on {args.project} and {args.zone}')
 
 
 def parse_env_config(args, tensorboard_config, system: SystemCharacteristics):