Updated the dtreeviz object to complice with dtreeviz v2.0.0 (#244)

* Updated the dtreeviz object to complice with dtreeviz v2.0.0 * Update dtreeviz import to be compatible with dtreeviz v2.0.0 and previous versions * Updated ShadowDecTree import method Co-authored-by: haizad <mhaizad.mtarik@petronas.com.my>
oegedijk · Dec 31, 2022 · f6998cb · f6998cb
1 parent 7d67e8f
commit f6998cb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/explainerdashboard/explainers.py b/explainerdashboard/explainers.py
@@ -25,7 +25,8 @@
 
 import shap
 
-from dtreeviz.trees import ShadowDecTree, dtreeviz
+from dtreeviz import *
+from dtreeviz.models.shadow_decision_tree import ShadowDecTree
 
 from sklearn.model_selection import KFold
 from sklearn.base import clone