Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Aug 29, 2024
1 parent 6fe71a5 commit 3f5366f
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 9 deletions.
11 changes: 9 additions & 2 deletions treeple/tree/manifold/_morf_splitter.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,17 @@ from cython.operator cimport dereference as deref
from libcpp.vector cimport vector

from ..._lib.sklearn.tree._criterion cimport Criterion
from ..._lib.sklearn.tree._utils cimport rand_int
from ..._lib.sklearn.tree._tree cimport ParentInfo
from .._utils cimport init_2dmemoryview, ravel_multi_index_cython, unravel_index_cython, fisher_yates_shuffle, compute_vectorized_indices_from_cartesian
from ..._lib.sklearn.tree._utils cimport rand_int
from .._sklearn_splitter cimport sort
from .._utils cimport (
compute_vectorized_indices_from_cartesian,
fisher_yates_shuffle,
init_2dmemoryview,
ravel_multi_index_cython,
unravel_index_cython,
)


cdef float64_t INFINITY = np.inf

Expand Down
8 changes: 1 addition & 7 deletions treeple/tree/tests/test_manifold.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,6 @@
import joblib
import numpy as np
import pytest
from numpy.testing import assert_array_equal
from sklearn import datasets
from numpy.testing import assert_almost_equal, assert_array_equal
from sklearn.base import is_classifier
from sklearn.datasets import load_iris, make_blobs
from sklearn.tree._tree import TREE_LEAF

from treeple.tree import (
ExtraObliqueDecisionTreeClassifier,
Expand Down Expand Up @@ -38,7 +33,6 @@
digits.target = digits.target[perm]



def test_splitters():
"""Test that splitters are picklable."""
X, y = digits.data, digits.target
Expand Down

0 comments on commit 3f5366f

Please sign in to comment.