Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "924c624a465933d4b3e3ca878a4c7cb5cae91bac"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_trainer_callback.py
View file @ 72d9e039
...@@ -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