Skip to content

Commit f3975e4

Browse files
committed
Adds a new DAG to test "Jobset Ready Healthiness" metric
- Adds an entire new DAG to test this metric automatically - Adds the "validate_replica_number" function - Adds a "READY_TPU" Jobset in jobset_util.py -Fix automatic cluster naming script -Created "validate_jobset_replica_number" funtion in jobset_util.py -Changed workload name to "IDLE_READY_TPU_20M" -Changed schedule time and added statement to only run on PROD environment Changed to new nodepool creation format Remove unused DAG flow
1 parent fdcca93 commit f3975e4

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

dags/tpu_observability/utils/jobset_util.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -555,9 +555,12 @@ def wait_for_jobset_ttr_to_be_found(node_pool: node_pool_info) -> bool:
555555
return len(time_series) > 0
556556

557557

558-
@task.sensor(poke_interval=30, timeout=600, mode="reschedule")
559-
def wait_for_jobset_status_occurrence(
560-
replica_type: str, job_name: str, node_pool: node_pool_info
558+
@task.sensor(poke_interval=30, timeout=900, mode="reschedule")
559+
def validate_jobset_replica_number(
560+
node_pool: node_pool_info,
561+
jobset_config: JobSet,
562+
replica_type: str,
563+
correct_replica_num: int,
561564
):
562565
"""
563566
A sensor which checks if are any jobset replicas in a status type.

0 commit comments

Comments
 (0)