Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions imblearn/_min_dependencies.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@
"numpydoc": ("1.5.0", "docs"),
"sphinxcontrib-bibtex": ("2.4.1", "docs"),
"pydata-sphinx-theme": ("0.13.3", "docs"),
"packaging":("14.1", "tests"),
}


Expand Down
4 changes: 2 additions & 2 deletions imblearn/tensorflow/tests/test_generator.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from distutils.version import LooseVersion
from packaging.version import parse

import numpy as np
import pytest
Expand Down Expand Up @@ -147,7 +147,7 @@ def accuracy(y_true, y_pred):

@pytest.mark.parametrize("sampler", [None, NearMiss(), RandomOverSampler()])
def test_balanced_batch_generator(data, sampler):
if LooseVersion(tf.__version__) < "2":
if parse(tf.__version__) < "2":
check_balanced_batch_generator_tf_1_X_X(data, sampler)
else:
check_balanced_batch_generator_tf_2_X_X_compat_1_X_X(data, sampler)
Expand Down