COSMIT get rid of deprecation warning in tree tests

larsmans · larsmans · commit 5847006f1197 · 2014-01-11T12:51:15.000+01:00
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)
