10000 sklearn.tree: Fix validation of class_names argument for plot_tree · scikit-learn/scikit-learn@7ac87cf · GitHub
[go: up one dir, main page]
Skip to content

Commit 7ac87cf

Browse files
2maz2maz
committed
sklearn.tree: Fix validation of class_names argument for plot_tree
1 parent 59048f9 commit 7ac87cf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎sklearn/tree/_export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def __repr__(self):
7979
"decision_tree": [DecisionTreeClassifier, DecisionTreeRegressor],
8080
"max_depth": [Interval(Integral, 0, None, closed="left"), None],
8181
"feature_names": [list, None],
82-
"class_names": [list, None],
82+
"class_names": ["array-like", "boolean", None],
8383
"label": [StrOptions({"all", "root", "none"})],
8484
"filled": ["boolean"],
8585
"impurity": ["boolean"],

0 commit comments

Comments
 (0)
0