Skip to content

Commit be4bd47

Browse files
committed
lint
1 parent 61e831f commit be4bd47

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mlinsights/mltree/tree_digitize.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,8 @@ def add_nodes(parent, i, j, is_left):
151151

152152
cl = DecisionTreeRegressor()
153153
cl.tree_ = tree
154-
cl.tree_.value[:, 0, 0] = numpy.array(values, dtype=numpy.float64)
154+
cl.tree_.value[:, 0, 0] = numpy.array( # pylint: disable=E1137
155+
values, dtype=numpy.float64)
155156
cl.n_outputs = 1
156157
cl.n_outputs_ = 1
157158
try:

0 commit comments

Comments
 (0)