Skip to content

Commit 727d980

Browse files
committed
Add "IDLE_READY_TPU_20M" workload to jobset_util.py
1 parent 65f603c commit 727d980

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
@@ -63,6 +63,15 @@ class Workload:
6363
Each workload is a JSON-escaped string, ready to be used as a shell argument.
6464
"""
6565

66+
IDLE_READY_TPU_20M = json.dumps(
67+
textwrap.dedent(
68+
"""
69+
sleep 1200
70+
"""
71+
),
72+
ensure_ascii=False,
73+
)
74+
6675
JAX_TPU_BENCHMARK = json.dumps(
6776
textwrap.dedent(
6877
"""

0 commit comments

Comments
 (0)