remove unused files from sklearn.tree._tree

uber · Dec 6, 2024 · d4f7f2c · d4f7f2c
1 parent 70df9ef
commit d4f7f2c
Show file tree

Hide file tree

Showing 11 changed files with 69 additions and 2,633 deletions.
diff --git a/causalml/inference/tree/_tree/__init__.py b/causalml/inference/tree/_tree/__init__.py
@@ -7,20 +7,8 @@
 
 from ._classes import (
     BaseDecisionTree,
-    DecisionTreeClassifier,
-    DecisionTreeRegressor,
-    ExtraTreeClassifier,
-    ExtraTreeRegressor,
 )
-from ._export import export_graphviz, export_text, plot_tree
 
 __all__ = [
     "BaseDecisionTree",
-    "DecisionTreeClassifier",
-    "DecisionTreeRegressor",
-    "ExtraTreeClassifier",
-    "ExtraTreeRegressor",
-    "export_graphviz",
-    "plot_tree",
-    "export_text",
 ]