black3

johnflem · May 10, 2023 · 71260a4 · 71260a4
1 parent 7cbff00
commit 71260a4
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/pycaret/internal/preprocess/iterative_imputer.py b/pycaret/internal/preprocess/iterative_imputer.py
@@ -453,21 +453,15 @@ def _impute_one_feature(
             if is_categorical_feat:
                 estimator = clone(self._cat_estimator)
                 if self.cat_estimator_prepare_for_categoricals_type:
-                    (
-                        estimator,
-                        prep_fit_params,
-                    ) = prepare_estimator_for_categoricals(
+                    (estimator, prep_fit_params,) = prepare_estimator_for_categoricals(
                         estimator,
                         categorical_indices,
                         preparation_type=self.cat_estimator_prepare_for_categoricals_type,
                     )
             else:
                 estimator = clone(self._num_estimator)
                 if self.num_estimator_prepare_for_categoricals_type:
-                    (
-                        estimator,
-                        prep_fit_params,
-                    ) = prepare_estimator_for_categoricals(
+                    (estimator, prep_fit_params,) = prepare_estimator_for_categoricals(
                         estimator,
                         categorical_indices,
                         preparation_type=self.num_estimator_prepare_for_categoricals_type,