Fix boolean

adam2392 · adam2392 · commit 4840d4e3e3ef · 2023-06-23T18:26:04.000-04:00
Signed-off-by: Adam Li &lt;adam2392@gmail.com&gt;
diff --git a/sklearn/ensemble/_forest.py b/sklearn/ensemble/_forest.py
@@ -221,7 +221,7 @@ class BaseForest(MultiOutputMixin, BaseEnsemble, metaclass=ABCMeta):
             None,
             Interval(Integral, 1, None, closed="left"),
         ],
-        "store_leaf_values": [bool],
+        "store_leaf_values": ["boolean"],
     }
 
     @abstractmethod
diff --git a/sklearn/tree/_classes.py b/sklearn/tree/_classes.py
@@ -123,7 +123,7 @@ class BaseDecisionTree(MultiOutputMixin, BaseEstimator, metaclass=ABCMeta):
         "max_leaf_nodes": [Interval(Integral, 2, None, closed="left"), None],
         "min_impurity_decrease": [Interval(Real, 0.0, None, closed="left")],
         "ccp_alpha": [Interval(Real, 0.0, None, closed="left")],
-        "store_leaf_values": [bool],
+        "store_leaf_values": ["boolean"],
     }
 
     @abstractmethod

Original file line number	Diff line number	Diff line change
`@@ -221,7 +221,7 @@ class BaseForest(MultiOutputMixin, BaseEnsemble, metaclass=ABCMeta):`
`221`	`221`	`None,`
`222`	`222`	`Interval(Integral, 1, None, closed="left"),`
`223`	`223`	`],`
`224`		`- "store_leaf_values": [bool],`
	`224`	`+ "store_leaf_values": ["boolean"],`
`225`	`225`	`}`
`226`	`226`
`227`	`227`	`@abstractmethod`
Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ class BaseDecisionTree(MultiOutputMixin, BaseEstimator, metaclass=ABCMeta):`
`123`	`123`	`"max_leaf_nodes": [Interval(Integral, 2, None, closed="left"), None],`
`124`	`124`	`"min_impurity_decrease": [Interval(Real, 0.0, None, closed="left")],`
`125`	`125`	`"ccp_alpha": [Interval(Real, 0.0, None, closed="left")],`
`126`		`- "store_leaf_values": [bool],`
	`126`	`+ "store_leaf_values": ["boolean"],`
`127`	`127`	`}`
`128`	`128`
`129`	`129`	`@abstractmethod`