src/transformers/trainer_callback.py/CallbackHandler/on_predict
class CallbackHandler:    def on_predict(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, metrics):
        return self.call_event("on_predict", args, state, control, metrics=metrics)