Skip to content

Commit

Permalink
Update CausalTree's base _tree module with sklearn 1.5.2 (#806)
Browse files Browse the repository at this point in the history
* update inference.tree._tree with sklearn 1.5.2
* fix lint errors
* remove unused files from sklearn.tree._tree
* fix the sklearn version to be >= 1.5.2, 1.6
* update the readthedocs config to use py39
  • Loading branch information
jeongyoonlee authored Jan 10, 2025
1 parent d369317 commit 27990a0
Show file tree
Hide file tree
Showing 22 changed files with 6,965 additions and 2,747 deletions.
2 changes: 1 addition & 1 deletion .readthedocs.yml
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ build:
python: "miniconda3-4.7"

conda:
environment: docs/environment-py38-rtd.yml
environment: docs/environment-py39-rtd.yml

python:
install:
Expand Down
12 changes: 11 additions & 1 deletion causalml/inference/tree/_tree/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,14 @@
"""
This part of tree structures definition was initially borrowed from
https://github.com/scikit-learn/scikit-learn/tree/1.0.2/sklearn/tree
https://github.com/scikit-learn/scikit-learn/tree/1.5.2/sklearn/tree
"""

"""Decision tree based models for classification and regression."""

from ._classes import (
BaseDecisionTree,
)

__all__ = [
"BaseDecisionTree",
]
Loading

0 comments on commit 27990a0

Please sign in to comment.