turning off different tests (temp)

bobisai · Oct 7, 2020 · 679fc13 · 679fc13
1 parent 11cf47e
commit 679fc13
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/tests/unit/test_lr_schedulers.py b/tests/unit/test_lr_schedulers.py
@@ -4,17 +4,12 @@
 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,

diff --git a/tests/unit/test_pipe.py b/tests/unit/test_pipe.py
@@ -17,6 +17,10 @@
 
 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)