fix: Fix test_checkpoiting.sh (GoogleCloudPlatform#1272)

RUEI4341 · web-flow · commit f361f4bdc5f1 · 2026-05-26T15:40:20.000+08:00
Update `maxtext_checkpointing` since AI-Hypercomputer/maxtext#3940 removed the `COLLECT_STACK_TRACE=${4}` in `tests/end_to_end/test_checkpointing.sh`
diff --git a/dags/multipod/maxtext_checkpointing.py b/dags/multipod/maxtext_checkpointing.py
@@ -58,7 +58,7 @@
       command = (
           "bash tests/end_to_end/test_checkpointing.sh "
           f" {run_name} {base_output_directory} {dataset_path}"
-          f" true tfds autoselected {async_checkpointing}",
+          f" tfds autoselected {async_checkpointing}",
       )
       maxtext_v4_configs_test = gke_config.get_gke_config(
           num_slices=1,

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@`
`58`	`58`	`command = (`
`59`	`59`	`"bash tests/end_to_end/test_checkpointing.sh "`
`60`	`60`	`f" {run_name} {base_output_directory} {dataset_path}"`
`61`		`- f" true tfds autoselected {async_checkpointing}",`
	`61`	`+ f" tfds autoselected {async_checkpointing}",`
`62`	`62`	`)`
`63`	`63`	`maxtext_v4_configs_test = gke_config.get_gke_config(`
`64`	`64`	`num_slices=1,`