From 25d2ca3db1f4103b7c4ced6d6235c58bd0ac61ec Mon Sep 17 00:00:00 2001
From: Diego Marvid <dmarvid@tryolabs.com>
Date: Fri, 5 Apr 2024 19:16:05 -0300
Subject: [PATCH] improve error handling in data container for explainer at
 training

---
 pipeline_lib/core/data_container.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/pipeline_lib/core/data_container.py b/pipeline_lib/core/data_container.py
index c66accf..178efe0 100644
--- 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