FIX remove classes modification

neurodata · Aug 22, 2023 · 9b4ff1b · 9b4ff1b
1 parent 01c1e55
commit 9b4ff1b
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/sktree/tree/_honest_tree.py b/sktree/tree/_honest_tree.py
@@ -423,9 +423,6 @@ def partial_fit(self, X, y, sample_weight=None, check_input=True, classes=None):
 
         _sample_weight[self.honest_indices_] = 0
 
-        if classes is None:
-            classes = np.unique(y).tolist()
-
         self.estimator_.partial_fit(
             X,
             y,
@@ -560,9 +557,6 @@ def _fit(
             # XXX: maybe error out if the tree_estimator is already fitted
             self.estimator_ = deepcopy(self.tree_estimator)
 
-        if classes is None:
-            classes = np.unique(y).tolist()
-
         # Learn structure on subsample
         self.estimator_._fit(
             X,