Merge pull request #3339 from AI-Hypercomputer:hengtaoguo-test

PiperOrigin-RevId: 879766115

Author: Google-ML-Automation

tests/unit/sharding_compare_test.py

@@ -33,8 +33,6 @@
 
 Transformer = models.transformer_as_linen
 
-pytestmark = [pytest.mark.cpu_only, pytest.mark.tpu_backend]
-
 
 def compute_checksum(d: dict) -> str:
   """Compute a checksum (SHA256) of a dictionary."""

tests/unit/train_compile_test.py

@@ -28,8 +28,6 @@
 from maxtext.trainers.pre_train.train_compile import main as train_compile_main
 from tests.utils.test_helpers import get_test_config_path
 
-pytestmark = [pytest.mark.external_training, pytest.mark.tpu_backend]
-
 
 class TrainCompile(unittest.TestCase):
   """Tests for the Ahead of Time Compilation functionality, train_compile.py"""