diff --git a/tests/functional_tests/test_groups/training/test_megatron_fsdp.py b/tests/functional_tests/test_groups/training/test_megatron_fsdp.py index 505268013a..b3f6744d34 100644 --- a/tests/functional_tests/test_groups/training/test_megatron_fsdp.py +++ b/tests/functional_tests/test_groups/training/test_megatron_fsdp.py @@ -352,7 +352,6 @@ def test_fsdp_pretrain_basic(self, tmp_path): clear_directories(tmp_path) @pytest.mark.run_only_on("GPU") - @pytest.mark.pleasefixme # Broken after TE2.14 bump def test_fsdp_pretrain_with_checkpoint(self, tmp_path): """ Test FSDP training with checkpoint saving using fsdp_dtensor format.