From: Christian Kastner Date: Mon, 17 Aug 2020 14:45:57 +0200 Subject: Switch to new scikit-learn parse_version --- imblearn/metrics/tests/test_classification.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/imblearn/metrics/tests/test_classification.py b/imblearn/metrics/tests/test_classification.py index e7972f0..326a8d2 100644 --- a/imblearn/metrics/tests/test_classification.py +++ b/imblearn/metrics/tests/test_classification.py @@ -14,7 +14,7 @@ from sklearn import datasets from sklearn import svm from sklearn.preprocessing import label_binarize -from sklearn.utils.fixes import np_version +from sklearn.utils.fixes import np_version, parse_version from sklearn.utils.validation import check_random_state from sklearn.utils._testing import assert_allclose from sklearn.utils._testing import assert_array_equal @@ -413,7 +413,7 @@ def test_classification_report_imbalanced_multiclass_with_unicode_label(): "red¢ 0.42 0.90 0.55 0.57 0.70 0.51 20 avg / total " "0.51 0.53 0.80 0.47 0.58 0.40 75" ) - if np_version[:3] < (1, 7, 0): + if np_version < parse_version('1.7.0'): with pytest.raises(RuntimeError, match="NumPy < 1.7.0"): classification_report_imbalanced(y_true, y_pred) else: