Fix tests of TrainerCallback (huggingface#10615)

* Fix tests of TrainerCallback * Update tests/test_trainer_callback.py Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
Iwontbecreative · Jul 15, 2021 · 3017fcb · 3017fcb
1 parent e648151
commit 3017fcb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_trainer_callback.py b/tests/test_trainer_callback.py
@@ -91,7 +91,7 @@ def get_trainer(self, a=0, b=0, train_len=64, eval_len=64, callbacks=None, disab
         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,