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