import assert_array_almost_equal from numpy

EpistasisLab · Jan 6, 2021 · f076653 · f076653
1 parent b1ba6d0
commit f076653
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/tests/one_hot_encoder_tests.py b/tests/one_hot_encoder_tests.py
@@ -29,14 +29,8 @@
 """
 
 import numpy as np
+from numpy.testing import assert_array_almost_equal
 import scipy.sparse
-
-try:
-    # deprecated as of 0.22
-    from sklearn.utils._testing import assert_array_almost_equal
-except ImportError:
-    from sklearn.utils.testing import assert_array_almost_equal
-
 from sklearn.datasets import load_iris, load_boston
 from sklearn.linear_model import LinearRegression
 from sklearn.pipeline import make_pipeline