Skip to content

Commit e0bf83b

Browse files
authored
remove six dependencies (#213)
1 parent 53ee3f6 commit e0bf83b

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

imblearn/base.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from collections import Counter
99

1010
import numpy as np
11-
from six import string_types
1211
from sklearn.base import BaseEstimator
1312
from sklearn.externals import six
1413
from sklearn.utils import check_X_y
@@ -204,7 +203,7 @@ def _validate_ratio(self):
204203
elif self.ratio <= 0:
205204
raise ValueError('Ratio cannot be negative.')
206205

207-
elif isinstance(self.ratio, string_types):
206+
elif isinstance(self.ratio, six.string_types):
208207
if self.ratio != 'auto':
209208
raise ValueError('Unknown string for the parameter ratio.')
210209
else:

imblearn/under_sampling/instance_hardness_threshold.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@
66
from collections import Counter
77

88
import numpy as np
9-
from six import string_types
109
import sklearn
1110
from sklearn.base import ClassifierMixin
1211
from sklearn.ensemble import RandomForestClassifier
12+
from sklearn.externals.six import string_types
1313

1414
from ..base import BaseBinarySampler
1515

0 commit comments

Comments
 (0)