fix v7x failure on fractional eval batch test

Author: NuojCheng

.github/workflows/ci_pipeline.yml

@@ -153,10 +153,7 @@ jobs:
 
   maxtext_jupyter_notebooks:
     needs: [analyze_code_changes, build_and_upload_maxtext_package]
-    if: |
-      always() &&
-      needs.analyze_code_changes.outputs.run_notebooks == 'true' &&
-      needs.build_and_upload_maxtext_package.result == 'success'
+    if: false  # Temporarily disabled — only tpu7x tests are enabled
     uses: ./.github/workflows/run_jupyter_notebooks.yml
     strategy:
         fail-fast: false
@@ -193,9 +190,7 @@ jobs:
   tpu-tests:
     name: ${{ matrix.flavor || 'TPU' }} tests
     needs: [gate_test_run]
-    if: |
-      always() &&
-      needs.gate_test_run.result == 'success'
+    if: false  # Temporarily disabled — only tpu7x tests are enabled
     uses: ./.github/workflows/run_tests_coordinator.yml
     strategy:
       fail-fast: false
@@ -210,9 +205,7 @@ jobs:
   maxtext_cpu_torch_reference_tests:
     name: cpu-torch-reference tests
     needs: [gate_test_run]
-    if: |
-      always() &&
-      needs.gate_test_run.result == 'success'
+    if: false  # Temporarily disabled — only tpu7x tests are enabled
     uses: ./.github/workflows/run_tests_coordinator.yml
     with:
       flavor: cpu-torch-reference
@@ -223,10 +216,10 @@ jobs:
   tpu7x-tests:
     name: TPU7X tests
     needs: [gate_test_run]
+    # REVERT_TPU7X: previously gated to: github.ref == 'refs/heads/main' && (github.event_name == 'schedule' || github.event_name == 'workflow_dispatch')
     if: |
       always() &&
-      needs.gate_test_run.result == 'success' &&
-      github.ref == 'refs/heads/main' && (github.event_name == 'schedule' || github.event_name == 'workflow_dispatch')
+      needs.gate_test_run.result == 'success'
     uses: ./.github/workflows/run_tests_coordinator.yml
     strategy:
       fail-fast: false
@@ -241,9 +234,7 @@ jobs:
   gpu-tests:
     name: ${{ matrix.flavor || 'GPU' }} tests
     needs: [gate_test_run]
-    if: |
-      always() &&
-      needs.gate_test_run.result == 'success'
+    if: false  # Temporarily disabled — only tpu7x tests are enabled
     strategy:
       fail-fast: false
       matrix:
@@ -258,9 +249,7 @@ jobs:
   cpu-tests:
     name: ${{ matrix.flavor || 'CPU' }} tests
     needs: [gate_test_run]
-    if: |
-      always() &&
-      needs.gate_test_run.result == 'success'
+    if: false  # Temporarily disabled — only tpu7x tests are enabled
     uses: ./.github/workflows/run_tests_coordinator.yml
     strategy:
       fail-fast: false
@@ -274,9 +263,7 @@ jobs:
 
   maxtext_tpu_pathways_unit_tests:
     needs: [gate_test_run]
-    if: |
-      always() &&
-      needs.gate_test_run.result == 'success'
+    if: false  # Temporarily disabled — only tpu7x tests are enabled
     uses: ./.github/workflows/run_pathways_tests.yml
     strategy:
         fail-fast: false
@@ -299,9 +286,7 @@ jobs:
 
   maxtext_tpu_pathways_integration_tests:
     needs: [gate_test_run]
-    if: |
-      always() &&
-      needs.gate_test_run.result == 'success'
+    if: false  # Temporarily disabled — only tpu7x tests are enabled
     uses: ./.github/workflows/run_pathways_tests.yml
     strategy:
         fail-fast: false

tests/integration/train_tests.py

@@ -689,6 +689,7 @@ def test_fractional_eval_batch_size(self):
         "steps=5",
         "enable_checkpointing=False",
         "enable_goodput_recording=False",
+        "max_target_length=4096",
         "dataset_type=synthetic",
         "remat_policy=minimal",
         "per_device_batch_size=1",