From 11cf47efe1729a68206a743a2e0f740f9d148234 Mon Sep 17 00:00:00 2001 From: Jeff Rasley Date: Tue, 6 Oct 2020 18:10:10 -0700 Subject: [PATCH] temporarily disable lr unit tests --- tests/unit/test_lr_schedulers.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/unit/test_lr_schedulers.py b/tests/unit/test_lr_schedulers.py index bf630b1c..2fa4a374 100755 --- a/tests/unit/test_lr_schedulers.py +++ b/tests/unit/test_lr_schedulers.py @@ -4,12 +4,17 @@ 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, -- GitLab