Fix dtreeviz decisiontree_view input handling (#321)

Author: oegedijk

RELEASE_NOTES.md

@@ -21,6 +21,7 @@
 - **XGBoost 3.1+ compatibility**: Fixed handling of string-formatted predictions and `base_score` values returned by XGBoost 3.1+. Added robust string-to-numeric conversion with proper regex fallback to handle various string formats (e.g., `'[3.2967056E1]'`, `'[8.563135E-2,7.169811E-1,1.9738752E-1]'`)
 - **XGBoost SHAP initialization**: Fixed `base_score` conversion in both `get_params()` and booster's internal JSON configuration to ensure SHAP TreeExplainer initializes correctly with XGBoost 3.1+
 - **RandomForest dtreeviz compatibility**: Fixed dtype handling for `y_train` (now uses `int` instead of `int16`) and observation array conversion for `predict_path()` to work with newer dtreeviz versions
+- **Dtreeviz decisiontree_view**: Ensure observations are passed as numpy arrays to avoid pandas label lookup errors when dtreeviz indexes features by integer position
 - **Pandas deprecation warnings**: Removed deprecated `pd.option_context("future.no_silent_downcasting")` and `copy=False` parameter from `.infer_objects()` calls
 - **Runtime warnings**: Fixed divide-by-zero warnings in classification plots and residuals plots (log-ratio calculations) by adding proper zero checks and using `np.divide()` with `where` parameter
 

explainerdashboard/explainers.py

@@ -4869,8 +4869,13 @@ def decisiontree_view(self, tree_idx, index, show_just_path=False):
 
         viz = DTreeVizAPI(self.shadow_trees[tree_idx])
 
+        x_row = self.get_X_row(index).squeeze()
+        if isinstance(x_row, pd.Series):
+            x = x_row.to_numpy()
+        else:
+            x = np.atleast_1d(np.asarray(x_row))
         return viz.view(
-            x=self.get_X_row(index).squeeze(),
+            x=x,
             fancy=False,
             show_node_labels=False,
             show_just_path=show_just_path,
@@ -5145,8 +5150,13 @@ def decisiontree_view(self, tree_idx, index, show_just_path=False, pos_label=Non
 
         viz = DTreeVizAPI(self.shadow_trees[tree_idx])
 
+        x_row = self.get_X_row(index).squeeze()
+        if isinstance(x_row, pd.Series):
+            x = x_row.to_numpy()
+        else:
+            x = np.atleast_1d(np.asarray(x_row))
         return viz.view(
-            x=self.get_X_row(index).squeeze(),
+            x=x,
             fancy=False,
             show_node_labels=False,
             show_just_path=show_just_path,

tests/test_dtreeviz_contracts.py

@@ -0,0 +1,8 @@
+import inspect
+
+import dtreeviz.trees as trees
+
+
+def test_dtreeviz_view_signature_includes_x():
+    sig = inspect.signature(trees.DTreeVizAPI.view)
+    assert "x" in sig.parameters

tests/test_randomforest_explainer.py

@@ -24,6 +24,14 @@ def test_rfclas_decisionpath_df(precalculated_rf_classifier_explainer, test_name
     assert isinstance(df, pd.DataFrame)
 
 
+def test_rfclas_decisiontree_view_contract(precalculated_rf_classifier_explainer):
+    precalculated_rf_classifier_explainer._graphviz_available = True
+    render = precalculated_rf_classifier_explainer.decisiontree_view(
+        tree_idx=0, index=0
+    )
+    assert isinstance(render, dtreeviz.utils.DTreeVizRender)
+
+
 def test_rfclas_plot_trees(precalculated_rf_classifier_explainer, test_names):
     fig = precalculated_rf_classifier_explainer.plot_trees(index=0)
     assert isinstance(fig, go.Figure)

tests/test_xgboost_treeviz.py

@@ -27,6 +27,14 @@ def test_xgbclas_decisionpath_df(precalculated_xgb_classifier_explainer, test_na
     assert isinstance(df, pd.DataFrame)
 
 
+def test_xgbclas_decisiontree_view_contract(precalculated_xgb_classifier_explainer):
+    precalculated_xgb_classifier_explainer._graphviz_available = True
+    render = precalculated_xgb_classifier_explainer.decisiontree_view(
+        tree_idx=0, index=0
+    )
+    assert isinstance(render, dtreeviz.utils.DTreeVizRender)
+
+
 def test_xgbclas_plot_trees(precalculated_xgb_classifier_explainer, test_names):
     fig = precalculated_xgb_classifier_explainer.plot_trees(index=0)
     assert isinstance(fig, go.Figure)