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