src/transformers/utils/notebook.py/NotebookProgressCallback/on_predict
class NotebookProgressCallback:    def on_predict(self, args, state, control, **kwargs):
        if self.prediction_bar is not None:
            self.prediction_bar.close()
        self.prediction_bar = None