From 72d9e039f9c78b9e6559456310ed2221192c0815 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Tue, 9 Mar 2021 16:25:32 -0500 Subject: [PATCH] Fix tests of TrainerCallback (#10615) * Fix tests of TrainerCallback * Update tests/test_trainer_callback.py Co-authored-by: Lysandre Debut --- tests/test_trainer_callback.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_trainer_callback.py b/tests/test_trainer_callback.py index 5c0af40f47..7f97766d31 100644 --- a/tests/test_trainer_callback.py +++ b/tests/test_trainer_callback.py @@ -91,7 +91,7 @@ class TrainerCallbackTest(unittest.TestCase): config = RegressionModelConfig(a=a, b=b) model = RegressionPreTrainedModel(config) - args = TrainingArguments(self.output_dir, disable_tqdm=disable_tqdm, **kwargs) + args = TrainingArguments(self.output_dir, disable_tqdm=disable_tqdm, report_to=[], **kwargs) return Trainer( model, args,