Skip to content

Commit 976fbe8

Browse files
committed
update tpu_info_format_validation_dags and tpu_info_metrics_dag DAG timeout to 45
1 parent aab432a commit 976fbe8

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

dags/common/scheduling_helper/scheduling_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,12 +58,12 @@ class DayOfWeek(enum.Enum):
5858
"multi_host_nodepool_rollback": DefaultTimeout,
5959
"node_pool_ttr_disk_size": dt.timedelta(minutes=90),
6060
"node_pool_ttr_update_label": dt.timedelta(minutes=90),
61-
"tpu_info_format_validation_dag": DefaultTimeout,
61+
"tpu_info_format_validation_dag": dt.timedelta(minutes=45),
6262
"tpu_sdk_monitoring_validation": DefaultTimeout,
6363
"jobset_ttr_kill_process": dt.timedelta(minutes=90),
6464
"jobset_uptime_validation": dt.timedelta(minutes=90),
6565
"jobset_ttr_drain_restart": DefaultTimeout,
66-
"tpu_info_metrics_verification": DefaultTimeout,
66+
"tpu_info_metrics_verification": dt.timedelta(minutes=45),
6767
"jobset_ttr_node_reboot": dt.timedelta(minutes=90),
6868
},
6969
TPU_INTERRUPTION_MOCK_CLUSTER.name: {

0 commit comments

Comments
 (0)