improve error handling in data container for explainer at training

tryolabs · Apr 5, 2024 · 25d2ca3 · 25d2ca3
1 parent 01b0fb1
commit 25d2ca3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pipeline_lib/core/data_container.py b/pipeline_lib/core/data_container.py
@@ -486,6 +486,11 @@ def explainer(self) -> Any:
         Any
             The explainer stored in the DataContainer.
         """
+        if self.is_train:
+            raise ValueError(
+                "Explainer is only available for prediction. Pipeline was executed on training"
+                " mode."
+            )
         return self["explainer"]
 
     @explainer.setter