Skip to content

Commit 736151c

Browse files
Remove collect_stack_trace config from checkpointing test (GoogleCloudPlatform#1271)
1 parent f361f4b commit 736151c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

dags/multipod/maxtext_gpu_end_to_end.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def run_maxtext_tests(dag: models.DAG):
181181
"grain-checkpoint-determinism": (
182182
"XLA_PYTHON_CLIENT_MEM_FRACTION=0.65 TF_FORCE_GPU_ALLOW_GROWTH=true "
183183
"bash tests/end_to_end/test_checkpointing.sh runner gs://runner-maxtext-logs "
184-
"gs://maxtext-dataset False c4-array_record dot_product",
184+
"gs://maxtext-dataset c4-array_record dot_product",
185185
1,
186186
),
187187
"checkpoint-compatibility": (

0 commit comments

Comments
 (0)