enable_goodput_recording=False

Author: Cjkkkk

MaxText/tests/integration_tests/train_tests.py

@@ -327,6 +327,7 @@ def test_gpu_cudnn_flash_jax(self):
         r"dataset_path=gs://maxtext-dataset",
         "steps=2",
         "enable_checkpointing=False",
+        "enable_goodput_recording=False",
         "attention=cudnn_flash_jax",
         "packing=False",
         rf"tokenizer_path={os.path.join(os.path.dirname(PKG_DIR), 'assets', 'tokenizer.llama2')}",