Add kwargs to CallbackHandler.on_evaluate signature

huggingface · Apr 20, 2021 · 8d4c85e · 8d4c85e
1 parent 9bb6e0d
commit 8d4c85e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/trainer_callback.py b/src/transformers/trainer_callback.py
@@ -358,9 +358,9 @@ def on_step_begin(self, args: TrainingArguments, state: TrainerState, control: T
     def on_step_end(self, args: TrainingArguments, state: TrainerState, control: TrainerControl):
         return self.call_event("on_step_end", args, state, control)
 
-    def on_evaluate(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, metrics):
+    def on_evaluate(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, metrics, **kwargs):
         control.should_evaluate = False
-        return self.call_event("on_evaluate", args, state, control, metrics=metrics)
+        return self.call_event("on_evaluate", args, state, control, metrics=metrics, **kwargs)
 
     def on_save(self, args: TrainingArguments, state: TrainerState, control: TrainerControl):
         control.should_save = False