Skip to content

fix: extend timeout for jobset_ttr_drain_restart to 90 minutes - #290

Open
lechipai0209 wants to merge 2 commits into
devfrom
tpu-obs/user/kim/fix_jobset_ttr_drain_restart_timeout
Open

fix: extend timeout for jobset_ttr_drain_restart to 90 minutes#290
lechipai0209 wants to merge 2 commits into
devfrom
tpu-obs/user/kim/fix_jobset_ttr_drain_restart_timeout

Conversation

@lechipai0209

@lechipai0209 lechipai0209 commented Jul 6, 2026

Copy link
Copy Markdown

Description

Change

Extend jobset_ttr_drain_restart timeout from 30mins to 90 mins.

Reason

In previous codebase, the timeout of this dag is 30 mins, so task wait_for_metric_up_load is often irrupted by dag timeout (this usually takes more than 45mins). The following clean_up tasks wouldn't be executed. The node pool and pods would be left in cluster until next test flow.

When the next test flow is starting, it will skip create_node_pool for existing node pool, and ensure_all_pods_running will reuse existing node pool but couldn't match resource with proper label.

Extending dag timeout could prevent the task from being stopped by dag timeout, so the clean up tasks can finish in every test flow.

Checklist

Before submitting this PR, please make sure (put X in square brackets):

  • I have performed a self-review of my code.
  • I have necessary comments in my code, particularly in hard-to-understand areas.
  • I have run one-shot tests and provided workload links above if applicable.
  • I have made or will make corresponding changes to the doc if needed.

@lechipai0209
lechipai0209 force-pushed the tpu-obs/user/kim/fix_jobset_ttr_drain_restart_timeout branch from 6bccef5 to 8838f7b Compare July 23, 2026 02:23
@lechipai0209 lechipai0209 removed the WIP label Jul 23, 2026
@alfredyu-cienet

Copy link
Copy Markdown
Collaborator

please finalize the PR description, we could at least mentioning "why"

extra information for you, PR description is essentially the same as commit message, which is very close to adding comments in code, where "why" is more important then "what"

"validate_interruption_count_gke_host_error": DefaultTimeout,
"validate_interruption_count_gce_defragmentation": DefaultTimeout,
"validate_interruption_count_gce_other": DefaultTimeout,
"validate_interruption_count_gce_migrate_on_hwsw_maintenance": DefaultTimeout,

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I don't think do line break for a config like this make sense,
so use # pylint: disable=line-too-long to disable it, and then enable back at the proper line

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

fixed

@lechipai0209
lechipai0209 force-pushed the tpu-obs/user/kim/fix_jobset_ttr_drain_restart_timeout branch from 8838f7b to 1a5e88e Compare July 23, 2026 08:19
@lechipai0209

Copy link
Copy Markdown
Author

please finalize the PR description, we could at least mentioning "why"

extra information for you, PR description is essentially the same as commit message, which is very close to adding comments in code, where "why" is more important then "what"

thanks. I add more information.

@alfredyu-cienet
alfredyu-cienet force-pushed the dev branch 2 times, most recently from 54f9a78 to 8e4ea82 Compare July 27, 2026 07:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants