fixes detailed plot bug with cats=None

oegedijk · May 5, 2021 · f0cba1e · f0cba1e
1 parent 3ebb2bb
commit f0cba1e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/explainerdashboard/explainers.py b/explainerdashboard/explainers.py
@@ -469,8 +469,12 @@ def X_cats(self):
     @property
     def X_merged(self, index=None):
         if index is None:
+            if self.X_cats.empty:
+                return self.X[self.merged_cols]
             return self.X.merge(self.X_cats, left_index=True, right_index=True)[self.merged_cols]
         else:
+            if self.X_cats.empty:
+                return self.X[index][self.merged_cols]
             return self.X[index].merge(self.X_cats[index], left_index=True, right_index=True)[self.merged_cols]
 
     @property