From 4975b05e89aaa29923b72c23b7b0f45118e4252f Mon Sep 17 00:00:00 2001 From: Min Xu <24926999+min-xu-ai@users.noreply.github.com> Date: Sat, 24 Sep 2022 18:59:24 -0700 Subject: [PATCH] reenable a test (#1081) Co-authored-by: Min Xu --- tests/nn/data_parallel/test_fsdp.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/nn/data_parallel/test_fsdp.py b/tests/nn/data_parallel/test_fsdp.py index e67c65fe0..d0df6dc60 100644 --- a/tests/nn/data_parallel/test_fsdp.py +++ b/tests/nn/data_parallel/test_fsdp.py @@ -481,7 +481,6 @@ def _get_model(self, group, config): return FullyShardedDataParallel(model, group, **config) -@pytest.mark.skip(reason="Recently flaky and not reproducible locally.") class TestSerialization(DistributedTest): @parameterized.expand([[False, False], [True, False], [True, True], [False, True]], name_func=rename_test) def test_pickle(self, mixed_precision, cpu_offload):