COSMIT get rid of deprecation warning in tree tests

Ben-Steele · Jan 11, 2014 · 5847006 · 5847006
1 parent b3a9da9
commit 5847006
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sklearn/tree/tests/test_tree.py b/sklearn/tree/tests/test_tree.py
@@ -28,7 +28,7 @@
 from sklearn import datasets
 from sklearn.utils.fixes import bincount
 
-from sklearn.preprocessing import balance_weights
+from sklearn.preprocessing._weights import _balance_weights
 
 
 CLF_CRITERIONS = ("gini", "entropy")
@@ -518,7 +518,7 @@ def test_unbalanced_iris():
     """Check class rebalancing."""
     unbalanced_X = iris.data[:125]
     unbalanced_y = iris.target[:125]
-    sample_weight = balance_weights(unbalanced_y)
+    sample_weight = _balance_weights(unbalanced_y)
 
     for name, TreeClassifier in CLF_TREES.items():
         clf = TreeClassifier(random_state=0)