Fix duplicate unit test skip

Summary: TSIA

Differential Revision: D75607800

Author: aporialiao

torchrec/distributed/tests/test_dynamic_sharding.py

@@ -498,14 +498,10 @@ def test_dynamic_sharding_ebc_cw(
 
 
 @skip_if_asan_class
-@unittest.skipIf(
-    torch.cuda.device_count() <= 3,
-    "Not enough GPUs, this test requires at least four GPUs",
-)
 class MultiRankDMPDynamicShardingTest(ModelParallelTestShared):
     @unittest.skipIf(
-        torch.cuda.device_count() <= 3,
-        "Not enough GPUs, this test requires at least four GPUs",
+        torch.cuda.device_count() <= 1,
+        "Not enough GPUs, this test requires at least two GPUs",
     )
     @given(  # pyre-ignore
         sharder_type=st.sampled_from(