fixes convert_str_tabs

oegedijk · Jan 12, 2021 · 98dbbb9 · 98dbbb9
1 parent fb652a7
commit 98dbbb9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/RELEASE_NOTES.md b/RELEASE_NOTES.md
@@ -16,7 +16,7 @@
     and how to sort the categories (alphabetical, by frequency or by mean abs shap)
 
 ### Bug Fixes
--
+- fixes bug where str tab indicators returned e.g. the old ImportancesTab instead of ImportancesComposite
 -
 
 ### Improvements

diff --git a/explainerdashboard/dashboards.py b/explainerdashboard/dashboards.py
@@ -658,19 +658,22 @@ def _store_params(self, no_store=None, no_attr=None, no_param=None):
     def _convert_str_tabs(self, component):
         if isinstance(component, str):
             if component == 'importances':
-                return ImportancesTab
+                return ImportancesComposite
             elif component == 'model_summary':
-                return ModelSummaryTab
+                if self.explainer.is_classifier:
+                    return ClassifierModelStatsComposite
+                else:
+                    return RegressionModelStatsComposite
             elif component == 'contributions':
-                return ContributionsTab
+                return IndividualPredictionsComposite
             elif component == 'whatif':
-                return WhatIfTab
+                return WhatIfComposite
             elif component == 'shap_dependence':
-                return ShapDependenceTab
+                return ShapDependenceComposite
             elif component == 'shap_interaction':
-                return ShapInteractionsTab
+                return ShapInteractionsComposite
             elif component == 'decision_trees':
-                return  DecisionTreesTab
+                return  DecisionTreesComposite
         return component
 
     @staticmethod