fix

Author: chiajunglien

dags/tpu_observability/jobset_healthiness_validation.py

@@ -113,7 +113,7 @@
           node_pool=cluster_info,
       )
 
-      startup = jobset.create_jobset_startup_group(
+      startup = jobset.create_jobset_startup_tasks(
           node_pool=cluster_info,
           jobset_config=jobset_config,
           workload_type=Workload.JAX_TPU_BENCHMARK,
@@ -239,7 +239,7 @@
           jobset_config,
           cluster_info,
           create_node_pool,
-          startup.task_group,
+          *startup.tasks,
           validate_running,
           suspend_action,
           validate_suspended,