Add utils.testing support for sklearn>=0.24.0

jmyrberg · Dec 29, 2020 · 9528f70 · 9528f70
1 parent 6bd0445
commit 9528f70
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_aakr.py b/tests/test_aakr.py
@@ -4,7 +4,11 @@
 import pytest
 
 from sklearn.datasets import load_linnerud
-from sklearn.utils.testing import assert_allclose
+
+try:  # scikit-learn < 0.24.0
+    from sklearn.utils.testing import assert_allclose
+except ModuleNotFoundError:  # scikit-learn >= 0.24.0
+    from sklearn.utils._testing import assert_allclose
 
 from aakr import AAKR