Refactor callback handling in Engine class by removing post-process…

…or and metrics callbacks. Simplify callback configuration to streamline the process and enhance maintainability.

src/anomalib/engine/engine.py

@@ -22,7 +22,6 @@
 from anomalib.deploy import CompressionType, ExportType
 from anomalib.models import AnomalibModule
 from anomalib.utils.path import create_versioned_dir
-from anomalib.visualization import ImageVisualizer
 
 logger = logging.getLogger(__name__)
 
@@ -258,7 +257,7 @@ def _setup_trainer(self, model: AnomalibModule) -> None:
             self._cache.update(model)
 
         # Setup anomalib callbacks to be used with the trainer
-        self._setup_anomalib_callbacks(model)
+        self._setup_anomalib_callbacks()
 
         # Temporarily set devices to 1 to avoid issues with multiple processes
         self._cache.args["devices"] = 1
@@ -267,7 +266,7 @@ def _setup_trainer(self, model: AnomalibModule) -> None:
         if self._trainer is None:
             self._trainer = Trainer(**self._cache.args)
 
-    def _setup_anomalib_callbacks(self, model: AnomalibModule) -> None:
+    def _setup_anomalib_callbacks(self) -> None:
         """Set up callbacks for the trainer."""
         _callbacks: list[Callback] = []
 
@@ -282,18 +281,6 @@ def _setup_anomalib_callbacks(self, model: AnomalibModule) -> None:
                 ),
             )
 
-        # Add the post-processor callback.
-        if isinstance(model.post_processor, Callback):
-            _callbacks.append(model.post_processor)
-
-        # Add the metrics callback.
-        if isinstance(model.evaluator, Callback):
-            _callbacks.append(model.evaluator)
-
-        # Add the image visualizer callback if it is passed by the user.
-        if not any(isinstance(callback, ImageVisualizer) for callback in self._cache.args["callbacks"]):
-            _callbacks.append(ImageVisualizer())
-
         _callbacks.append(TimerCallback())
 
         # Combine the callbacks, and update the trainer callbacks.