diff --git a/sklearn_pandas/pipeline.py b/sklearn_pandas/pipeline.py index e8aae87..cde2fea 100644 --- a/sklearn_pandas/pipeline.py +++ b/sklearn_pandas/pipeline.py @@ -1,4 +1,3 @@ -import six from sklearn.pipeline import _name_estimators, Pipeline from sklearn.utils import tosequence @@ -59,7 +58,7 @@ def __init__(self, steps): def _pre_transform(self, X, y=None, **fit_params): fit_params_steps = dict((step, {}) for step, _ in self.steps) - for pname, pval in six.iteritems(fit_params): + for pname, pval in fit_params.items(): step, param = pname.split('__', 1) fit_params_steps[step][param] = pval Xt = X