SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix tests of TrainerCallback (#10615)

Browse Source 
* Fix tests of TrainerCallback

* Update tests/test_trainer_callback.py

Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
This commit is contained in:
Sylvain Gugger
2021-03-09 16:25:32 -05:00
committed by GitHub
parent 0d909f6bd8
commit 72d9e039f9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_trainer_callback.py
View File

@@ -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,