Skip to content

Commit

Permalink
Update submodule
Browse files Browse the repository at this point in the history
Signed-off-by: Adam Li <[email protected]>
  • Loading branch information
adam2392 committed Aug 16, 2023
1 parent eafb318 commit cfabd4e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sktree/tree/_classes.py
Original file line number Diff line number Diff line change
Expand Up @@ -1630,7 +1630,7 @@ def _build_tree(
)

if self.data_dims is None:
self.data_dims_ = np.array((1, X.shape[1]), dtype=np.int8)
self.data_dims_ = np.array((1, X.shape[1]), dtype=np.intp)
else:
if np.prod(self.data_dims) != X.shape[1]:
raise RuntimeError(f"Data dimensions {self.data_dims} do not match {X.shape[1]}.")
Expand Down

0 comments on commit cfabd4e

Please sign in to comment.