Make predict() close progress bars after finishing (#17952) (#18078)

* Make Trainer.predict call on_evaluate (#17952)

* Add on_predict

* Small fix

* Small and different fix

* Add tests

tests/trainer/test_trainer_callback.py

@@ -66,6 +66,9 @@ class MyTestTrainerCallback(TrainerCallback):
     def on_evaluate(self, args, state, control, **kwargs):
         self.events.append("on_evaluate")
 
+    def on_predict(self, args, state, control, **kwargs):
+        self.events.append("on_predict")
+
     def on_save(self, args, state, control, **kwargs):
         self.events.append("on_save")