def require_deepspeed(test_case): """ Decorator marking a test that requires deepspeed """ if not is_deepspeed_available(): return unittest.skip("test requires deepspeed")(test_case) else: return test_case
with open(path) as f: return json.load(f) # a candidate for testing_utils def require_deepspeed(test_case): """ Decorator marking a test that requires deepspeed """ if not is_deepspeed_available(): return unittest.skip("test requires deepspeed")(test_case) else: return test_case if is_deepspeed_available(): from deepspeed.utils import logger as deepspeed_logger # noqa from transformers.integrations import deepspeed_config, is_deepspeed_zero3_enabled # noqa ZERO2 = "zero2" ZERO3 = "zero3" stages = [ZERO2, ZERO3] @require_deepspeed @require_torch_gpu class TrainerIntegrationDeepSpeed(TestCasePlus, TrainerIntegrationCommon): """ This class is for testing directly via get_regression_trainer