update dtreeviz to 2.0.0

mljar · Dec 30, 2022 · 7f51765 · 7f51765
1 parent eff1641
commit 7f51765
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -8,7 +8,7 @@ catboost>=0.24.4
 joblib>=1.0.1
 tabulate>=0.8.7
 matplotlib>=3.2.2
-dtreeviz>=1.3.3
+dtreeviz>=2.0.0
 shap>=0.36.0
 seaborn>=0.11.1
 wordcloud>=1.8.1

diff --git a/supervised/algorithms/decision_tree.py b/supervised/algorithms/decision_tree.py
@@ -20,7 +20,7 @@
 logger.setLevel(LOG_LEVEL)
 
 from sklearn.tree import _tree
-from dtreeviz.trees import dtreeviz
+from dtreeviz import * 
 from supervised.utils.subsample import subsample
 
 

diff --git a/supervised/algorithms/knn.py b/supervised/algorithms/knn.py
@@ -20,8 +20,6 @@
 logger.setLevel(LOG_LEVEL)
 
 
-from dtreeviz.trees import dtreeviz
-
 KNN_ROWS_LIMIT = 1000
Original file line number	Diff line number	Diff line change
Expand Up		@@ -20,8 +20,6 @@
		logger.setLevel(LOG_LEVEL)


		from dtreeviz.trees import dtreeviz

		KNN_ROWS_LIMIT = 1000


Expand Down