turning off different tests (temp)

tests/unit/test_lr_schedulers.py

@@ -4,17 +4,12 @@ import argparse
 import pytest
 import json
 import os
-import pytest
 from common import distributed_test
 from simple_model import SimpleModel, SimpleOptimizer, random_dataloader, args_from_dict
 from deepspeed.runtime.lr_schedules import LR_RANGE_TEST, ONE_CYCLE, WARMUP_LR, WARMUP_DECAY_LR
 from deepspeed.runtime.lr_schedules import WARMUP_MIN_LR, WARMUP_MAX_LR, WARMUP_NUM_STEPS, TOTAL_NUM_STEPS
 from deepspeed.runtime.lr_schedules import CYCLE_MIN_LR, CYCLE_MAX_LR
 
-pytest.skip(
-    "skipping until we can figure out what's causing these to hang inside our CI",
-    allow_module_level=True)
-
 
 @pytest.mark.parametrize("scheduler_type,params",
                          [(WARMUP_LR,

tests/unit/test_pipe.py

@@ -17,6 +17,10 @@ from deepspeed.runtime.pipe.module import LayerSpec
 
 from common import distributed_test
 
+pytest.skip(
+    "skipping until we can figure out what's causing these to hang inside our CI",
+    allow_module_level=True)
+
 
 def rel_diff(A, B):
     return abs(A - B) / abs(A)