src/transformers/trainer_callback.py/TrainerCallback/on_prediction_step
class TrainerCallback:    def on_prediction_step(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, **kwargs):
        """
        Event called after a prediction step.
        """
        pass