Unverified Commit 72d9e039 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix tests of TrainerCallback (#10615)



* Fix tests of TrainerCallback

* Update tests/test_trainer_callback.py
Co-authored-by: default avatarLysandre Debut <lysandre@huggingface.co>
parent 0d909f6b
...@@ -91,7 +91,7 @@ class TrainerCallbackTest(unittest.TestCase): ...@@ -91,7 +91,7 @@ class TrainerCallbackTest(unittest.TestCase):
config = RegressionModelConfig(a=a, b=b) config = RegressionModelConfig(a=a, b=b)
model = RegressionPreTrainedModel(config) 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( return Trainer(
model, model,
args, args,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment