From 633062639bfd6be15abc072aaf7e18bce355f426 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Thu, 23 Feb 2023 13:22:25 -0800 Subject: [PATCH] [deepspeed tests] fix issues introduced by #21700 (#21769) * [deepspeed tests] fix issues introduced by #21700 * fix * fix --- tests/deepspeed/test_deepspeed.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/deepspeed/test_deepspeed.py b/tests/deepspeed/test_deepspeed.py index 60cec456c3..9b203dfd7b 100644 --- a/tests/deepspeed/test_deepspeed.py +++ b/tests/deepspeed/test_deepspeed.py @@ -19,10 +19,12 @@ import json import os import unittest from copy import deepcopy +from functools import partial import datasets from parameterized import parameterized +import tests.trainer.test_trainer from tests.trainer.test_trainer import TrainerIntegrationCommon # noqa from transformers import AutoModel, TrainingArguments, is_torch_available, logging from transformers.deepspeed import HfDeepSpeedConfig, is_deepspeed_available, unset_hf_deepspeed_config @@ -49,9 +51,11 @@ if is_torch_available(): from tests.trainer.test_trainer import ( # noqa RegressionModelConfig, RegressionPreTrainedModel, - get_regression_trainer, ) + # hack to restore original logging level pre #21700 + get_regression_trainer = partial(tests.trainer.test_trainer.get_regression_trainer, log_level="info") + set_seed(42)