Merge pull request #2877 from AI-Hypercomputer:flaky2

PiperOrigin-RevId: 848221566

Author: Google-ML-Automation

codecov.yml

@@ -24,6 +24,8 @@
 # During PRs, the 'scheduled' flag is carried forward from the last full run on 'main' to keep the score stable.
 
 # Exclude non-source code, deprecated and experimental folders from coverage tracking
+codecov: 
+ token: 35742a22-fb1f-4839-97ff-b54da5588689
 ignore:
   - "src/MaxText/assets"
   - "src/MaxText/configs"

tests/decode_tests.py

@@ -136,6 +136,7 @@ def test_gpu_pdb_lt_1(self):
 
   @pytest.mark.tpu_only
   @pytest.mark.scheduled_only
+  @pytest.mark.skip(reason="Flaky test. Disabled in b/470982884")
   def test_decode_greedy_sampling(self):
     config = DecodeTests.CONFIGS["decode_sampling"] + DecodeTests.SAMPLING_STRATEGY_CONFIG["greedy"]
     captured_out = run_decoding(config)
@@ -160,6 +161,7 @@ def test_decode_nucleus_sampling(self):
 
   @pytest.mark.tpu_only
   @pytest.mark.scheduled_only
+  @pytest.mark.skip(reason="Flaky test. Disabled in b/470982884")
   def test_decode_topk_sampling(self):
     config = DecodeTests.CONFIGS["decode_sampling"] + DecodeTests.SAMPLING_STRATEGY_CONFIG["topk"]
     captured_out = run_decoding(config)