riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix a bug for CallbackHandler.callback_list (#8052)

Browse Source 
* Fix callback_list

* Add test

Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>

* Fix test

Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
This commit is contained in:
Harutaka Kawamura 2020-10-27 23:37:04 +09:00 committed by GitHub
parent 8e28c327fc
commit 7bff0af0a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/trainer_callback.py
View File

@ -325,7 +325,7 @@ class CallbackHandler(TrainerCallback):
@property @property
def callback_list(self): def callback_list(self):
return "\n".join(self.callbacks) return "\n".join(cb.__class__.__name__ for cb in self.callbacks)
def on_init_end(self, args: TrainingArguments, state: TrainerState, control: TrainerControl): def on_init_end(self, args: TrainingArguments, state: TrainerState, control: TrainerControl):
return self.call_event("on_init_end", args, state, control) return self.call_event("on_init_end", args, state, control)

7
tests/test_trainer_callback.py
View File

@ -221,3 +221,10 @@ class TrainerCallbackTest(unittest.TestCase):
trainer.train() trainer.train()
events = trainer.callback_handler.callbacks[-2].events events = trainer.callback_handler.callbacks[-2].events
self.assertEqual(events, self.get_expected_events(trainer)) self.assertEqual(events, self.get_expected_events(trainer))
# warning should be emitted for duplicated callbacks
with unittest.mock.patch("transformers.trainer_callback.logger.warn") as warn_mock:
trainer = self.get_trainer(
callbacks=[MyTestTrainerCallback, MyTestTrainerCallback],
)
assert str(MyTestTrainerCallback) in warn_mock.call_args[0][0]