Skip to content

Commit 0973a13

Browse files
committed
Add "IDLE_READY_TPU_20M" workload to jobset_util.py
1 parent d85d904 commit 0973a13

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

dags/tpu_observability/utils/jobset_util.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,15 @@ class Workload:
4343
Each workload is a JSON-escaped string, ready to be used as a shell argument.
4444
"""
4545

46+
IDLE_READY_TPU_20M = json.dumps(
47+
textwrap.dedent(
48+
"""
49+
sleep 1200
50+
"""
51+
),
52+
ensure_ascii=False,
53+
)
54+
4655
JAX_TPU_BENCHMARK = json.dumps(
4756
textwrap.dedent(
4857
"""

0 commit comments

Comments
 (0)